Searched refs:process_batch_callback (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler.h | 186 process_batch_callback,
217 process_batch_callback, in Create()
236 process_batch_callback, in Create()
|
| D | shared_batch_scheduler.h | 164 process_batch_callback,
234 Env* env, ProcessBatchCallback process_batch_callback,
395 process_batch_callback, in AddQueue()
418 options, options_.env, process_batch_callback, in AddQueue()
508 Env* env, ProcessBatchCallback process_batch_callback, in Queue() argument
512 process_batch_callback_(process_batch_callback), in Queue()
|
| D | basic_batch_scheduler_benchmark_test.cc | 167 auto process_batch_callback = in ResetState() local
172 scheduler_options_, process_batch_callback, &scheduler_)); in ResetState()
296 auto process_batch_callback = in ResetState() local
301 scheduler_options_, process_batch_callback, &scheduler_)); in ResetState()
|
| 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()
|
| D | adaptive_shared_batch_scheduler.h | 138 BatchProcessor process_batch_callback,
365 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue() argument
380 queues_and_callbacks_[asbs_queue_raw] = process_batch_callback; in AddQueue()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 662 auto process_batch_callback = [this](std::unique_ptr<Batch> batch) { in LookupOrCreateBatcherQueue() local
670 process_batch_callback, &new_queue)); in LookupOrCreateBatcherQueue()
|