Searched refs:cuda_scatter_gather_base_kernel (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ScatterGatherKernel.cu | 160 struct cuda_scatter_gather_base_kernel { struct
472 cuda_scatter_gather_base_kernel</*is_scatter_like=*/false>()( in gather_cuda_kernel()
480 cuda_scatter_gather_base_kernel<>()( in scatter_cuda_kernel()
495 cuda_scatter_gather_base_kernel</*is_scatter_like=*/true, /*cast_to_opaque=*/false>()( in scatter_add_cuda_kernel()
507 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_cuda_kernel()
511 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_cuda_kernel()
524 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_two_cuda_kernel()
529 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_two_cuda_kernel()
533 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_two_cuda_kernel()
537 cuda_scatter_gather_base_kernel<true, false>()(self, dim, index, src, in scatter_reduce_two_cuda_kernel()
[all …]
|