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 | 269 process_batch_callback,
300 process_batch_callback, in Create()
332 process_batch_callback, in Create()
|
| D | basic_batch_scheduler_benchmark_test.cc | 163 auto process_batch_callback = in ResetState() local
168 scheduler_options_, process_batch_callback, &scheduler_)); in ResetState()
292 auto process_batch_callback = in ResetState() local
297 scheduler_options_, process_batch_callback, &scheduler_)); in ResetState()
|
| D | shared_batch_scheduler.h | 201 process_batch_callback,
275 Env* env, ProcessBatchCallback process_batch_callback,
460 process_batch_callback, in AddQueue()
502 options, options_.env, process_batch_callback, in AddQueue()
592 Env* env, ProcessBatchCallback process_batch_callback, in Queue() argument
596 process_batch_callback_(process_batch_callback), in Queue()
|
| 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 | 164 BatchProcessor process_batch_callback,
407 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue() argument
431 queues_and_callbacks_[asbs_queue_raw] = process_batch_callback; in AddQueue()
|
| D | batch_resource_base.cc | 737 auto process_batch_callback = [this](std::unique_ptr<BatchT> batch) { in LookupOrCreateBatcherQueue() local
746 process_batch_callback, &new_queue)); in LookupOrCreateBatcherQueue()
749 adaptive_batcher_queue_options_, process_batch_callback, &new_queue)); in LookupOrCreateBatcherQueue()
|