Searched refs:compute_num_of_partial_segments (Results 1 – 1 of 1) sorted by relevance
193 __global__ void compute_num_of_partial_segments(const index_t *partials_per_segment, const index_t … in compute_num_of_partial_segments() function281 compute_num_of_partial_segments<index_t><<<1, 1, 0, c10::cuda::getCurrentCUDAStream()>>>( in embedding_backward_cuda_kernel()