Searched defs:segment_ends (Results 1 – 1 of 1) sorted by relevance
63 TIndex* __restrict__ unique_input_inds, TIndex* __restrict__ segment_ends) { in ExtractFirstOccurrenceIndicesKernel()91 TIndex* segment_ends) { in ExtractFirstOccurrenceIndices()108 const TIndex* __restrict__ segment_ends, T* __restrict__ output, in GatherOutputsAndInvertPermutationKernel()129 const TIndex* segment_ends, T* output, in GatherOutputsAndInvertPermutation()344 Tensor segment_ends; in ComputeAsync() local