Searched refs:sorted_input_unique_ids (Results 1 – 1 of 1) sorted by relevance
62 const TIndex* __restrict__ sorted_input_unique_ids, in ExtractFirstOccurrenceIndicesKernel() argument65 TIndex sorted_input_unique_id = sorted_input_unique_ids[i]; in ExtractFirstOccurrenceIndicesKernel()66 if (i == 0 || sorted_input_unique_id != sorted_input_unique_ids[i - 1]) { in ExtractFirstOccurrenceIndicesKernel()89 const TIndex* sorted_input_unique_ids, in ExtractFirstOccurrenceIndices() argument99 sorted_input_unique_ids, unique_input_inds, in ExtractFirstOccurrenceIndices()146 const TIndex* __restrict__ sorted_input_unique_ids, in LookupAndScatterUniqueIdsKernel() argument151 inv_sorted_unique_perm[sorted_input_unique_ids[i]]; in LookupAndScatterUniqueIdsKernel()160 const TIndex* sorted_input_unique_ids, in LookupAndScatterUniqueIds() argument170 sorted_input_unique_ids, inv_sorted_unique_perm, idx); in LookupAndScatterUniqueIds()300 Tensor sorted_input_unique_ids; in ComputeAsync() local[all …]