Home
last modified time | relevance | path

Searched refs:max_task_size (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler.h278 size_t max_task_size() const override { in max_task_size() function
279 return shared_scheduler_queue_->max_task_size(); in max_task_size()
Dserial_device_batch_scheduler.h226 size_t max_task_size() const override { return options_.max_batch_size; } in max_task_size() function
391 static_cast<double>((*best_it)->queue()->max_task_size()); in ProcessBatches()
396 static_cast<double>((*it)->queue()->max_task_size()); in ProcessBatches()
Dbasic_batch_scheduler_test.cc76 EXPECT_EQ(10, scheduler->max_task_size()); in TEST()
Dshared_batch_scheduler.h300 size_t max_task_size() const { return options_.input_batch_size_limit; } in max_task_size() function
410 size_t max_task_size() const override { return queue_->max_task_size(); } in max_task_size() function
Dbatch_scheduler.h195 virtual size_t max_task_size() const = 0;
Dadaptive_shared_batch_scheduler.h284 size_t max_task_size() const override { return options_.max_batch_size; } in max_task_size() function
482 static_cast<double>((*it)->queue()->max_task_size()); in MaybeScheduleNextBatch()
Dshared_batch_scheduler_test.cc431 EXPECT_EQ(2, queue->max_task_size()); in TEST()