Searched defs:uniq_size (Results 1 – 2 of 2) sorted by relevance
60 int64_t input_size, int64_t uniq_size, in ExtractFirstOccurrenceIndicesKernel()87 int64_t uniq_size, in ExtractFirstOccurrenceIndices()105 int64_t uniq_size, const T* __restrict__ input, in GatherOutputsAndInvertPermutationKernel()125 Status GatherOutputsAndInvertPermutation(const GPUDevice& d, int64_t uniq_size, in GatherOutputsAndInvertPermutation()333 int64 uniq_size = (*last_idx_host.data()) + 1; in ComputeAsync() local
145 int64_t uniq_size; in Compute() local