Searched refs:scatter_add_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
491 void scatter_add_cuda_kernel(const Tensor& self, int64_t dim, const Tensor& index, const Tensor& sr… in scatter_add_cuda_kernel() function568 REGISTER_DISPATCH(scatter_add_stub, &scatter_add_cuda_kernel);