Searched refs:flatten_indices_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
20 Tensor flatten_indices_cuda_kernel(const Tensor& indices, IntArrayRef size) { in flatten_indices_cuda_kernel() function26 REGISTER_CUDA_DISPATCH(flatten_indices_stub, &flatten_indices_cuda_kernel);