Searched refs:BatchProcessor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | adaptive_shared_batch_scheduler.h | 160 using BatchProcessor = std::function<void(std::unique_ptr<Batch<TaskType>>)>; variable
164 BatchProcessor process_batch_callback,
180 BatchProcessor callback, bool is_express);
209 std::unordered_map<const internal::ASBSQueue<TaskType>*, BatchProcessor>
407 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
533 AdaptiveSharedBatchScheduler<TaskType>::BatchProcessor callback, in CallbackWrapper()
|
| D | serial_device_batch_scheduler.h | 113 using BatchProcessor = std::function<void(std::unique_ptr<Batch<TaskType>>)>; variable
117 BatchProcessor process_batch_callback,
154 std::unordered_map<const internal::SDBSQueue<TaskType>*, BatchProcessor>
333 const QueueOptions& options, BatchProcessor process_batch_callback, in AddQueue()
|