Searched refs:max_partial_segment (Results 1 – 1 of 1) sorted by relevance
287 auto max_partial_segment = numel / NROWS_PER_THREAD + max_segments; in embedding_backward_cuda_kernel() local291 auto partial_segment_offset = at::empty({max_partial_segment}, orig_indices.options()); in embedding_backward_cuda_kernel()305 const int grid = ceil_div(max_partial_segment*stride_warped, block); in embedding_backward_cuda_kernel()318 auto grad_weight_per_segment = at::empty({max_partial_segment, stride}, op); in embedding_backward_cuda_kernel()