Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dserial_device_batch_scheduler.h38 class SDBSBatch; variable
140 void AddBatch(const internal::SDBSBatch<TaskType>* batch);
151 std::vector<const internal::SDBSBatch<TaskType>*> batches_ TF_GUARDED_BY(mu_);
224 void ReleaseBatch(const SDBSBatch<TaskType>* batch);
232 SDBSBatch<TaskType>* current_batch_ TF_GUARDED_BY(mu_) = nullptr;
241 class SDBSBatch : public Batch<TaskType> {
243 SDBSBatch(SDBSQueue<TaskType>* queue, int64 creation_time_micros) in SDBSBatch() function
246 ~SDBSBatch() override {} in ~SDBSBatch()
255 TF_DISALLOW_COPY_AND_ASSIGN(SDBSBatch);
354 const internal::SDBSBatch<TaskType>* batch) { in AddBatch()
[all …]