Searched refs:cpu_scatter_reduce_expanded_index (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ScatterGatherKernel.cpp | 689 void cpu_scatter_reduce_expanded_index(const Tensor& self, const Tensor& index, const Tensor& src, … in cpu_scatter_reduce_expanded_index() function
854 …cpu_scatter_reduce_expanded_index<scalar_t, ReductionType::SUM>(self, index, src, /*include_self*/… in scatter_add_expanded_index_kernel()
864 cpu_scatter_reduce_expanded_index<scalar_t, reduce>(self, index, src, include_self); in scatter_reduce_expanded_index_kernel()
|