Searched refs:unique_by_key (Results 1 – 4 of 4) sorted by relevance
92 thrust::pair<thrust_ptr, thrust_ptr> newEnd = thrust::unique_by_key(policy, in _coalesce_sparse_cuda()
77 num_out = thrust::unique_by_key(policy, data, data + num_inp, range_ptr, equal).first - data; in compute_unique()
245 cuda::cub::unique_by_key( in embedding_backward_cuda_kernel()
163 inline void unique_by_key( in unique_by_key() function