Searched defs:num_unique_indices (Results 1 – 1 of 1) sorted by relevance
201 const int64_t *num_unique_indices) { in renorm_kernel()361 auto num_unique_indices = at::empty({}, indices.options().dtype(kLong)); in embedding_renorm_cuda_() local