Searched refs:cpu_scatter_gather_base_kernel (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ScatterGatherKernel.cpp | 163 struct cpu_scatter_gather_base_kernel { struct
877 cpu_scatter_gather_base_kernel</*is_scatter_like=*/false>()( in gather_cpu_kernel()
883 cpu_scatter_gather_base_kernel<>()( in scatter_cpu_kernel()
888 cpu_scatter_gather_base_kernel<>()( in scatter_fill_cpu_kernel()
893 cpu_scatter_gather_base_kernel<>()( in scatter_add_cpu_kernel()
902 cpu_scatter_gather_base_kernel<>()(self, dim, index, src, in scatter_reduce_cpu_kernel()
906 cpu_scatter_gather_base_kernel<>()(self, dim, index, src, in scatter_reduce_cpu_kernel()
918 cpu_scatter_gather_base_kernel<>()(self, dim, index, src, in scatter_reduce_two_cpu_kernel()
922 cpu_scatter_gather_base_kernel<>()(self, dim, index, src, in scatter_reduce_two_cpu_kernel()
926 cpu_scatter_gather_base_kernel<>()(self, dim, index, src, in scatter_reduce_two_cpu_kernel()
[all …]
|