Searched refs:gather_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
471 void gather_cuda_kernel(const Tensor& result, const Tensor& self, int64_t dim, const Tensor& index)… in gather_cuda_kernel() function565 REGISTER_DISPATCH(gather_stub, &gather_cuda_kernel);