Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dadaptive_shared_batch_scheduler.h46 class ASBSBatch; variable
179 void CallbackWrapper(const internal::ASBSBatch<TaskType>* batch,
195 void AddBatch(const internal::ASBSBatch<TaskType>* batch);
206 std::vector<const internal::ASBSBatch<TaskType>*> batches_ TF_GUARDED_BY(mu_);
282 void ReleaseBatch(const ASBSBatch<TaskType>* batch);
297 ASBSBatch<TaskType>* current_batch_ TF_GUARDED_BY(mu_) = nullptr;
306 class ASBSBatch : public Batch<TaskType> {
308 ASBSBatch(ASBSQueue<TaskType>* queue, int64 creation_time_micros, in ASBSBatch() function
315 ~ASBSBatch() override {} in ~ASBSBatch()
330 TF_DISALLOW_COPY_AND_ASSIGN(ASBSBatch);
[all …]