Home
last modified time | relevance | path

Searched refs:SDBSQueue (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dserial_device_batch_scheduler.h41 class SDBSQueue; variable
132 friend class internal::SDBSQueue<TaskType>;
143 void RemoveQueue(const internal::SDBSQueue<TaskType>* queue);
154 std::unordered_map<const internal::SDBSQueue<TaskType>*, BatchProcessor>
201 class SDBSQueue : public BatchScheduler<TaskType> {
206 SDBSQueue(std::shared_ptr<SerialDeviceBatchScheduler<TaskType>> scheduler,
209 ~SDBSQueue() override;
236 TF_DISALLOW_COPY_AND_ASSIGN(SDBSQueue);
243 SDBSBatch(SDBSQueue<TaskType>* queue, int64 creation_time_micros) in SDBSBatch()
248 SDBSQueue<TaskType>* queue() const { return queue_; } in queue()
[all …]