Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dadaptive_shared_batch_scheduler.h151 absl::optional<int> max_input_task_size = absl::nullopt; member
455 if (options.max_input_task_size.has_value()) { in AddQueue()
456 if (options.max_input_task_size.value() < options.max_batch_size) { in AddQueue()
460 options.max_input_task_size.value(), " and max_batch_size as ", in AddQueue()
731 if (options_.max_input_task_size.has_value() && in Schedule()
732 (size > options_.max_input_task_size.value())) { in Schedule()
735 options_.max_input_task_size.value()); in Schedule()
H A Dbatch_resource_base.cc367 batcher_queue_options.max_input_task_size = in GetAdaptiveBatcherQueueOptions()