Searched refs:scatter_reduce_two_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
914 void scatter_reduce_two_cpu_kernel(const Tensor& self, const int64_t dim, const Tensor& index, in scatter_reduce_two_cpu_kernel() function964 REGISTER_DISPATCH(scatter_reduce_two_stub, &scatter_reduce_two_cpu_kernel);