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