Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dserial_device_batch_scheduler.h140 void AddBatch(const internal::SDBSBatch<TaskType>* batch);
353 void SerialDeviceBatchScheduler<TaskType>::AddBatch( in AddBatch() function
514 if (new_batch != nullptr) scheduler_->AddBatch(new_batch); in Schedule()
Dadaptive_shared_batch_scheduler.h195 void AddBatch(const internal::ASBSBatch<TaskType>* batch);
436 void AdaptiveSharedBatchScheduler<TaskType>::AddBatch( in AddBatch() function
704 scheduler_->AddBatch(batch); in Schedule()