Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dadaptive_shared_batch_scheduler.h86 delete batch_thread_pool_; in ~AdaptiveSharedBatchScheduler() local
240 thread::ThreadPool* batch_thread_pool_; variable
434 batch_thread_pool_ = new thread::ThreadPool( in AdaptiveSharedBatchScheduler()
438 batch_thread_pool_ = options.thread_pool; in AdaptiveSharedBatchScheduler()
512 batch_thread_pool_->Schedule(std::bind( in MaybeScheduleNextBatchFIFO()
532 batch_thread_pool_->Schedule( in MaybeScheduleClosedBatchesLockedFIFO()
582 batch_thread_pool_->Schedule( in MaybeScheduleNextBatch()
611 batch_thread_pool_->Schedule( in MaybeScheduleClosedBatchesLocked()
H A 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()
444 batch_thread_pool_->Schedule(std::bind( in ProcessBatches()