Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc139 AdaptiveBatcherT::Options adaptive_shared_batch_scheduler_options, in Create() argument
147 adaptive_shared_batch_scheduler_options, &batcher)); in Create()
336 adaptive_shared_batch_scheduler_options; in ComputeAsync() local
337 adaptive_shared_batch_scheduler_options.thread_pool_name = in ComputeAsync()
339 adaptive_shared_batch_scheduler_options.num_batch_threads = in ComputeAsync()
341 adaptive_shared_batch_scheduler_options.thread_pool = in ComputeAsync()
355 adaptive_shared_batch_scheduler_options.min_in_flight_batches_limit = in ComputeAsync()
357 adaptive_shared_batch_scheduler_options.initial_in_flight_batches_limit = in ComputeAsync()
359 adaptive_shared_batch_scheduler_options.batches_to_average_over = in ComputeAsync()
361 adaptive_shared_batch_scheduler_options.fifo_scheduling = true; in ComputeAsync()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc158 AdaptiveBatcherT::Options adaptive_shared_batch_scheduler_options, in Create() argument
166 adaptive_shared_batch_scheduler_options, &batcher)); in Create()
305 adaptive_shared_batch_scheduler_options; in ComputeAsync() local
306 adaptive_shared_batch_scheduler_options.thread_pool_name = in ComputeAsync()
308 adaptive_shared_batch_scheduler_options.num_batch_threads = in ComputeAsync()
310 adaptive_shared_batch_scheduler_options.thread_pool = in ComputeAsync()
324 adaptive_shared_batch_scheduler_options.min_in_flight_batches_limit = in ComputeAsync()
326 adaptive_shared_batch_scheduler_options.initial_in_flight_batches_limit = in ComputeAsync()
328 adaptive_shared_batch_scheduler_options.batches_to_average_over = in ComputeAsync()
330 adaptive_shared_batch_scheduler_options.fifo_scheduling = true; in ComputeAsync()
[all …]