Searched refs:scatter_scalar_reduce_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
548 void scatter_scalar_reduce_cuda_kernel(const Tensor& self, const int64_t dim, const Tensor& index, in scatter_scalar_reduce_cuda_kernel() function570 REGISTER_DISPATCH(scatter_scalar_reduce_stub, &scatter_scalar_reduce_cuda_kernel);