Searched refs:cumsum_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
91 void cumsum_cuda_kernel(const Tensor& result, const Tensor& self, int64_t dim) { in cumsum_cuda_kernel() function112 REGISTER_CUDA_DISPATCH(cumsum_stub, &cumsum_cuda_kernel);