Home
last modified time | relevance | path

Searched refs:CPUKernelLauncher (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseBinaryOpIntersectionKernel.cpp13 struct CPUKernelLauncher { struct
129 _sparse_binary_op_intersection_kernel_out<CPUKernelLauncher, CPUValueSelectionMulKernel>( in mul_sparse_sparse_out_cpu_kernel()
140 _sparse_binary_op_intersection_kernel_out<CPUKernelLauncher, CPUValueRhsProjKernel>( in sparse_mask_intersection_out_cpu_kernel()
152 _sparse_binary_op_intersection_kernel_out<CPUKernelLauncher, CPUValueLhsProjKernel>( in sparse_mask_projection_out_cpu_kernel()
H A DFlattenIndicesKernel.cpp13 struct CPUKernelLauncher { struct
20 return _flatten_indices<CPUKernelLauncher>(indices, size); in flatten_indices_cpu_kernel()