Searched defs:sorted_input_inds (Results 1 – 1 of 1) sorted by relevance
61 const TIndex* __restrict__ sorted_input_inds, in ExtractFirstOccurrenceIndicesKernel()88 const TIndex* sorted_input_inds, in ExtractFirstOccurrenceIndices()145 int64_t input_size, const TIndex* sorted_input_inds, in LookupAndScatterUniqueIdsKernel()159 const TIndex* sorted_input_inds, in LookupAndScatterUniqueIds()272 Tensor sorted_input_inds; in ComputeAsync() local