Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dadaptive_shared_batch_scheduler.h49 class ASBSQueue; variable
174 friend class internal::ASBSQueue<TaskType>;
198 void RemoveQueue(const internal::ASBSQueue<TaskType>* queue);
209 std::unordered_map<const internal::ASBSQueue<TaskType>*, BatchProcessor>
259 class ASBSQueue : public BatchScheduler<TaskType> {
264 ASBSQueue(std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>> scheduler,
267 ~ASBSQueue() override;
301 TF_DISALLOW_COPY_AND_ASSIGN(ASBSQueue);
308 ASBSBatch(ASBSQueue<TaskType>* queue, int64 creation_time_micros, in ASBSBatch()
317 ASBSQueue<TaskType>* queue() const { return queue_; } in queue()
[all …]