Home
last modified time | relevance | path

Searched refs:kMaxBatchSize (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler_benchmark_test.cc344 const int kMaxBatchSize = 100; in RunThroughputBenchmark() local
345 scheduler_options.max_batch_size = kMaxBatchSize; in RunThroughputBenchmark()
393 const int kMaxBatchSize = 100; in RunLatencyBenchmark() local
394 scheduler_options.max_batch_size = kMaxBatchSize; in RunLatencyBenchmark()
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc199 const int kMaxBatchSize = 128; in Compute() local
232 std::forward_as_tuple(factors_mat.rows(), kMaxBatchSize)); in Compute()
266 if (num_batched == kMaxBatchSize) { in Compute()