Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dserial_device_batch_scheduler.h158 std::unique_ptr<thread::ThreadPool> batch_thread_pool_; variable
312 batch_thread_pool_.reset(new thread::ThreadPool( in SerialDeviceBatchScheduler()
315 batch_thread_pool_->Schedule( in SerialDeviceBatchScheduler()
328 batch_thread_pool_.reset(); in ~SerialDeviceBatchScheduler()
443 batch_thread_pool_->Schedule(std::bind( in ProcessBatches()
Dadaptive_shared_batch_scheduler.h85 batch_thread_pool_.reset(); in ~AdaptiveSharedBatchScheduler()
215 std::unique_ptr<thread::ThreadPool> batch_thread_pool_; variable
398 batch_thread_pool_.reset(new thread::ThreadPool( in AdaptiveSharedBatchScheduler()
401 batch_thread_pool_.reset(new thread::ThreadPool(options.thread_pool)); in AdaptiveSharedBatchScheduler()
494 batch_thread_pool_->Schedule( in MaybeScheduleNextBatch()
519 batch_thread_pool_->Schedule( in MaybeScheduleClosedBatchesLocked()