Home
last modified time | relevance | path

Searched refs:input_batch_size_limit (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler.h144 size_t input_batch_size_limit = 1000; member
187 int first_output_task_size, int input_batch_size_limit,
300 size_t max_task_size() const { return options_.input_batch_size_limit; } in max_task_size()
309 return options_.input_batch_size_limit; in max_execution_batch_size()
462 if (options.input_batch_size_limit == 0) { in AddQueue()
465 options.input_batch_size_limit); in AddQueue()
487 (options.input_batch_size_limit < options.max_execution_batch_size)) { in AddQueue()
492 options.enable_large_batch_splitting, options.input_batch_size_limit, in AddQueue()
625 if ((*task)->size() > options_.input_batch_size_limit) { in ScheduleWithoutSplit()
628 options_.input_batch_size_limit); in ScheduleWithoutSplit()
[all …]
Dshared_batch_scheduler_test.cc100 queue_options.input_batch_size_limit = 10; in TEST()
158 queue_options.input_batch_size_limit = 10; in TEST()
220 queue_options.input_batch_size_limit = 4; in TEST()
276 queue_options.input_batch_size_limit = 10; in TEST()
321 queue_options.input_batch_size_limit = 100; in TEST()
374 queue_options.input_batch_size_limit = 10; in TEST()
426 queue_options.input_batch_size_limit = 2; in TEST()
497 queue_options.input_batch_size_limit = 10; in TEST()
553 queue_options.input_batch_size_limit = 10; in TEST()
Dbasic_batch_scheduler.h244 int first_output_task_size, int input_batch_size_limit,
318 shared_scheduler_queue_options.input_batch_size_limit = in Create()
Dbatch_resource_base.cc232 batcher_queue_options.input_batch_size_limit = max_batch_size; in GetBatcherQueueOptions()