Searched refs:partial_segment_offset (Results 1 – 1 of 1) sorted by relevance
291 auto partial_segment_offset = at::empty({max_partial_segment}, orig_indices.options()); in embedding_backward_cuda_kernel() local294 partial_segment_offset.mutable_data_ptr<index_t>(), in embedding_backward_cuda_kernel()329 partial_segment_offset.const_data_ptr<index_t>(), in embedding_backward_cuda_kernel()339 partial_segment_offset.const_data_ptr<index_t>(), in embedding_backward_cuda_kernel()