Searched defs:per_batch_counts (Results 1 – 1 of 1) sorted by relevance
41 Status OutputSparse(const BatchedMap<T>& per_batch_counts, int64_t num_values, in OutputSparse()150 auto per_batch_counts = BatchedMap<W>(num_batch_elements); in Compute() local273 auto per_batch_counts = BatchedMap<W>(num_batches); in Compute() local366 auto per_batch_counts = BatchedMap<W>(num_batches); in Compute() local