Home
last modified time | relevance | path

Searched defs:process_batch_callback (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dshared_batch_scheduler.h552 std::function<void(std::unique_ptr<Batch<TaskType>>)> in AddQueue()
573 std::function<void(std::unique_ptr<Batch<TaskType>>)> in AddQueueAfterRewritingOptions()
760 Env* env, ProcessBatchCallback process_batch_callback, in Queue()
H A Dbasic_batch_scheduler_benchmark_test.cc115 auto process_batch_callback = in ThroughputBenchmark() local
209 auto process_batch_callback = in LatencyBenchmark() local
H A Dbasic_batch_scheduler.h305 std::function<void(std::unique_ptr<Batch<TaskType>>)> in Create()
H A Dshared_batch_scheduler_test.cc108 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback) { in CreateQueue()
843 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback = in CreateQueues() local
H A Dserial_device_batch_scheduler.h333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
H A Dadaptive_shared_batch_scheduler.h444 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
H A Dbatch_resource_base.cc875 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue() local