Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dadaptive_shared_batch_scheduler.h140 absl::optional<int> max_input_task_size = absl::nullopt; member
418 if (options.max_input_task_size.has_value()) { in AddQueue()
419 if (options.max_input_task_size.value() < options.max_batch_size) { in AddQueue()
423 options.max_input_task_size.value(), " and max_batch_size as ", in AddQueue()
628 if (options_.max_input_task_size.has_value() && in Schedule()
629 (size > options_.max_input_task_size.value())) { in Schedule()
632 options_.max_input_task_size.value()); in Schedule()
Dbatch_resource_base.cc263 batcher_queue_options.max_input_task_size = in GetAdaptiveBatcherQueueOptions()