Searched refs:compute_counts_uniq (Results 1 – 1 of 1) sorted by relevance
1485 static std::vector<index_t> compute_counts_uniq( in compute_counts_uniq() function1548 compute_counts_uniq(num_weights, indices_data, numel, counts); in _embedding_bag_dense_backward_cpu_sum_mean()