Searched refs:total_values (Results 1 – 2 of 2) sorted by relevance
34 int total_values = 0; in OutputSparse() local37 total_values += per_batch_count.size(); in OutputSparse()43 0, TensorShape({total_values, inner_dim}), &indices)); in OutputSparse()47 context->allocate_output(1, TensorShape({total_values}), &values)); in OutputSparse()
42 uint64_t total_values = 1ull << (std::numeric_limits<T>::digits + in UniformBucketTest() local46 static_cast<T>(total_values % static_cast<uint64_t>(bucket_count)); in UniformBucketTest()