Searched defs:partials_per_segment (Results 1 – 1 of 1) sorted by relevance
63 const index_t *partials_per_segment, in krn_partial_segment_offset()193 __global__ void compute_num_of_partial_segments(const index_t *partials_per_segment, const index_t … in compute_num_of_partial_segments()258 auto partials_per_segment = at::empty({max_segments}, orig_indices.options()); in embedding_backward_cuda_kernel() local