Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler.h394 std::function<void(std::unique_ptr<Batch<TaskType>>)> in AddQueue()
508 Env* env, ProcessBatchCallback process_batch_callback, in Queue()
Dbasic_batch_scheduler.h216 std::function<void(std::unique_ptr<Batch<TaskType>>)> in Create()
Dbasic_batch_scheduler_benchmark_test.cc167 auto process_batch_callback = in ResetState() local
296 auto process_batch_callback = in ResetState() local
Dadaptive_shared_batch_scheduler.h365 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
Dserial_device_batch_scheduler.h333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc662 auto process_batch_callback = [this](std::unique_ptr<Batch> batch) { in LookupOrCreateBatcherQueue() local