Searched refs:sorted_unique_input_inds (Results 1 – 1 of 1) sorted by relevance
106 const TIndex* __restrict__ sorted_unique_input_inds, in GatherOutputsAndInvertPermutationKernel() argument111 output[i] = input[sorted_unique_input_inds[i]]; in GatherOutputsAndInvertPermutationKernel()127 const TIndex* sorted_unique_input_inds, in GatherOutputsAndInvertPermutation() argument138 d.stream(), uniq_size, input, sorted_unique_input_inds, in GatherOutputsAndInvertPermutation()360 Tensor sorted_unique_input_inds; in ComputeAsync() local362 AllocateTemp(context, uniq_size, &sorted_unique_input_inds, in ComputeAsync()421 sorted_unique_input_inds = Tensor(); in ComputeAsync()