Searched defs:inv_sorted_unique_perm (Results 1 – 1 of 1) sorted by relevance
109 TIndex* __restrict__ inv_sorted_unique_perm, TIndex* __restrict__ count) { in GatherOutputsAndInvertPermutationKernel()130 TIndex* inv_sorted_unique_perm, in GatherOutputsAndInvertPermutation()147 const TIndex* __restrict__ inv_sorted_unique_perm, in LookupAndScatterUniqueIdsKernel()161 const TIndex* inv_sorted_unique_perm, in LookupAndScatterUniqueIds()393 Tensor inv_sorted_unique_perm; in ComputeAsync() local