Home
last modified time | relevance | path

Searched defs:KernelLauncher (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DValidateCompressedIndicesCommon.h172 struct KernelLauncher { struct
174 static void launch( in launch()
182 static void launch(TensorIteratorBase& iter, const func_t& f) { in launch()
237 using KernelLauncher = KernelLauncher<kernel_t, vec_kernel_t>; in _validate_compressed_sparse_indices_kernel() local
H A DSparseBinaryOpIntersectionCommon.h70 struct KernelLauncher { struct
72 static void launch(TensorIteratorBase& iter, const func_t& f) { in launch()
150 using KernelLauncher = KernelLauncher<kernel_t>; variable
H A DFlattenIndicesCommon.h29 struct KernelLauncher { struct
31 static void launch(TensorIteratorBase& iter, const func_t& f) { in launch()