Home
last modified time | relevance | path

Searched refs:CUDAKernelLauncher (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSparseBinaryOpIntersectionKernel.cu14 struct CUDAKernelLauncher { struct
177 _sparse_binary_op_intersection_kernel_out<CUDAKernelLauncher, CUDAValueSelectionMulKernel>( in mul_sparse_sparse_out_cuda_kernel()
188 _sparse_binary_op_intersection_kernel_out<CUDAKernelLauncher, CUDAValueRhsProjKernel>( in sparse_mask_intersection_out_cuda_kernel()
200 _sparse_binary_op_intersection_kernel_out<CUDAKernelLauncher, CUDAValueLhsProjKernel>( in sparse_mask_projection_out_cuda_kernel()
H A DFlattenIndicesKernel.cu14 struct CUDAKernelLauncher { struct
21 return _flatten_indices<CUDAKernelLauncher>(indices, size); in flatten_indices_cuda_kernel()
H A DValidateCompressedIndicesKernel.cu10 struct CUDAKernelLauncher { struct
25 validate_compressed_sparse_indices_kernel<CUDAKernelLauncher>( in _validate_compressed_sparse_indices_cuda()