Searched defs:sorted_unique_perm (Results 1 – 1 of 1) sorted by relevance
107 const TIndex* __restrict__ sorted_unique_perm, in GatherOutputsAndInvertPermutationKernel()128 const TIndex* sorted_unique_perm, in GatherOutputsAndInvertPermutation()366 Tensor sorted_unique_perm; in ComputeAsync() local