Searched defs: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
165 void operator()(const Tensor& self, int64_t dim, in operator ()()
262 void operator()(const Tensor& self, int64_t dim, in operator ()()
358 void operator()(const Tensor& self, int64_t dim, in operator ()()
454 void operator()(const Tensor& self, int64_t dim, in operator ()()
549 void operator()(const Tensor& self, int64_t dim, in operator ()()
|