Searched refs:per_batch_counts (Results 1 – 4 of 4) sorted by relevance
32 Status OutputSparse(const BatchedMap<T>& per_batch_counts, int num_values, in OutputSparse() argument35 int num_batches = per_batch_counts.size(); in OutputSparse()36 for (const auto& per_batch_count : per_batch_counts) { in OutputSparse()53 const auto& per_batch_count = per_batch_counts[b]; in OutputSparse()128 auto per_batch_counts = BatchedMap<W>(num_batch_elements); in Compute() local140 per_batch_counts[b][value] = 1; in Compute()142 per_batch_counts[b][value] += weight_values(i); in Compute()144 per_batch_counts[b][value]++; in Compute()155 OP_REQUIRES_OK(context, OutputSparse<W>(per_batch_counts, num_output_values, in Compute()209 auto per_batch_counts = BatchedMap<W>(num_batches); in Compute() local[all …]
14 auto per_batch_counts = BatchedMap<W>(num_batches);22 per_batch_counts[batch_idx - 1][value] = 1;29 never be 1, hence there will be no hashmap at index 0 in `per_batch_counts`.
14 auto per_batch_counts = BatchedMap<W>(num_batches);
28 per_batch_counts[batch][value] += weight_values(idx);