Searched refs:process_batch_callback (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | shared_batch_scheduler.h | 224 process_batch_callback,
244 process_batch_callback,
311 Env* env, ProcessBatchCallback process_batch_callback,
553 process_batch_callback, in AddQueue()
566 return AddQueueAfterRewritingOptions(rewrite_options, process_batch_callback, in AddQueue()
574 process_batch_callback, in AddQueueAfterRewritingOptions()
622 options, options_.env, process_batch_callback, in AddQueueAfterRewritingOptions()
760 Env* env, ProcessBatchCallback process_batch_callback, in Queue() argument
765 process_batch_callback_(process_batch_callback), in Queue()
|
| H A D | basic_batch_scheduler_benchmark_test.cc | 115 auto process_batch_callback = in ThroughputBenchmark() local
120 scheduler_options_, process_batch_callback, &scheduler_)); in ThroughputBenchmark()
209 auto process_batch_callback = in LatencyBenchmark() local
214 scheduler_options_, process_batch_callback, &scheduler_)); in LatencyBenchmark()
|
| H A D | basic_batch_scheduler.h | 275 process_batch_callback,
306 process_batch_callback, in Create()
339 process_batch_callback, in Create()
|
| H A D | shared_batch_scheduler_test.cc | 108 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback) { in CreateQueue() argument
111 scheduler->AddQueue(queue_options, process_batch_callback, &queue)); in CreateQueue()
843 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback = in CreateQueues() local
860 process_batch_callback)); in CreateQueues()
870 process_batch_callback)); in CreateQueues()
880 process_batch_callback)); in CreateQueues()
|
| H A D | serial_device_batch_scheduler.h | 117 BatchProcessor process_batch_callback,
333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue() argument
348 queues_and_callbacks_[SDBS_queue_raw] = process_batch_callback; in AddQueue()
|
| H A D | adaptive_shared_batch_scheduler.h | 177 BatchProcessor process_batch_callback,
444 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue() argument
468 queues_and_callbacks_[asbs_queue_raw] = process_batch_callback; in AddQueue()
|
| H A D | batch_resource_base.cc | 875 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue() local
884 process_batch_callback, &new_queue)); in LookupOrCreateBatcherQueue()
887 adaptive_batcher_queue_options_, process_batch_callback, &new_queue)); in LookupOrCreateBatcherQueue()
|