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/
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()
Dbasic_batch_scheduler_benchmark_test.cc115 auto process_batch_callback = in ThroughputBenchmark() local
209 auto process_batch_callback = in LatencyBenchmark() local
Dbasic_batch_scheduler.h305 std::function<void(std::unique_ptr<Batch<TaskType>>)> in Create()
Dserial_device_batch_scheduler.h333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
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
Dadaptive_shared_batch_scheduler.h444 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
Dbatch_resource_base.cc875 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue() local