Searched defs:num_of_partial_segments_ptr (Results 1 – 1 of 1) sorted by relevance
164 const index_t *segment_sizes_offsets, const int64_t *num_of_partial_segments_ptr, in sum_and_scatter()280 … int64_t *num_of_partial_segments_ptr = num_of_partial_segments_tensor.mutable_data_ptr<int64_t>(); in embedding_backward_cuda_kernel() local