Searched refs:num_of_segments_tensor (Results 1 – 1 of 1) sorted by relevance
235 auto num_of_segments_tensor = at::empty({}, grad.options().dtype(kLong)); in embedding_backward_cuda_kernel() local236 int64_t *num_of_segments_ptr = num_of_segments_tensor.mutable_data_ptr<int64_t>(); in embedding_backward_cuda_kernel()