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.h195 size_t max_task_size() const override { in max_task_size() function
196 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()
Dshared_batch_scheduler.h253 size_t max_task_size() const { return options_.max_batch_size; } in max_task_size() function
345 size_t max_task_size() const override { return queue_->max_task_size(); } in max_task_size() function
Dbasic_batch_scheduler_test.cc76 EXPECT_EQ(10, scheduler->max_task_size()); in TEST()
Dbatch_scheduler.h184 virtual size_t max_task_size() const = 0;
Dadaptive_shared_batch_scheduler.h257 size_t max_task_size() const override { return options_.max_batch_size; } in max_task_size() function
426 static_cast<double>((*it)->queue()->max_task_size()); in MaybeScheduleNextBatch()
Dshared_batch_scheduler_test.cc431 EXPECT_EQ(2, queue->max_task_size()); in TEST()