Home
last modified time | relevance | path

Searched refs:batcher_queue_options (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc231 BatcherT::QueueOptions batcher_queue_options; in GetBatcherQueueOptions() local
232 batcher_queue_options.input_batch_size_limit = max_batch_size; in GetBatcherQueueOptions()
233 batcher_queue_options.max_enqueued_batches = max_enqueued_batches; in GetBatcherQueueOptions()
234 batcher_queue_options.batch_timeout_micros = batch_timeout_micros; in GetBatcherQueueOptions()
235 batcher_queue_options.enable_large_batch_splitting = in GetBatcherQueueOptions()
238 batcher_queue_options.split_input_task_func = in GetBatcherQueueOptions()
247 batcher_queue_options.max_execution_batch_size = max_batch_size; in GetBatcherQueueOptions()
249 batcher_queue_options.max_execution_batch_size = in GetBatcherQueueOptions()
254 return batcher_queue_options; in GetBatcherQueueOptions()
262 AdaptiveBatcherT::QueueOptions batcher_queue_options; in GetAdaptiveBatcherQueueOptions() local
[all …]
Dbatch_resource_base.h118 const BatcherT::QueueOptions& batcher_queue_options, in BatchResourceBase() argument
122 batcher_queue_options_(batcher_queue_options), in BatchResourceBase()
129 const AdaptiveBatcherT::QueueOptions& batcher_queue_options, in BatchResourceBase() argument
133 adaptive_batcher_queue_options_(batcher_queue_options), in BatchResourceBase()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc132 const BatcherT::QueueOptions& batcher_queue_options, in BatchResource() argument
136 std::move(batcher), batcher_queue_options, in BatchResource()
144 const AdaptiveBatcherT::QueueOptions& batcher_queue_options, in BatchResource() argument
148 std::move(batcher), batcher_queue_options, in BatchResource()