Searched refs:unpack_pivots_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
106 void unpack_pivots_cuda_kernel(TensorIterator& iter, const int64_t dim_size, const int64_t max_pivo… in unpack_pivots_cuda_kernel() function113 unpack_pivots_cuda_kernel(sub_iter, dim_size, max_pivot); in unpack_pivots_cuda_kernel()142 REGISTER_DISPATCH(unpack_pivots_stub, &unpack_pivots_cuda_kernel);