Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler_benchmark_test.cc115 auto process_batch_callback = in ThroughputBenchmark() local
209 auto process_batch_callback = in LatencyBenchmark() local
Dbasic_batch_scheduler.h306 std::function<void(std::unique_ptr<Batch<TaskType>>)> in Create()
Dshared_batch_scheduler.h537 std::function<void(std::unique_ptr<Batch<TaskType>>)> in AddQueue()
687 Env* env, ProcessBatchCallback process_batch_callback, in Queue()
Dserial_device_batch_scheduler.h333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
Dadaptive_shared_batch_scheduler.h432 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
Dshared_batch_scheduler_test.cc107 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback) { in CreateQueue()
860 internal::Queue<FakeTask>::ProcessBatchCallback process_batch_callback = in CreateQueues() local
Dbatch_resource_base.cc832 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue() local