Searched refs:prefetch_thread_ (Results 1 – 3 of 3) sorted by relevance
128 prefetch_thread_(std::move(iterator), num_consumers_) { in RoundRobinTaskRunner()151 TF_RETURN_IF_ERROR(prefetch_thread_.FillBuffer(wait_us, buffer_)); in PrepareFullRound()170 TF_RETURN_IF_ERROR(prefetch_thread_.FillBuffer(/*wait_us=*/-1, buffer_)); in PreparePartialRound()198 TF_RETURN_IF_ERROR(prefetch_thread_.GetStatus()); in PrepareRound()207 return prefetch_thread_.GetStatus(); in PrepareRound()
166 PrefetchThread prefetch_thread_; variable
432 if (!prefetch_thread_) { in EnsurePrefetchThreadStarted()435 prefetch_thread_ = ctx->StartThread( in EnsurePrefetchThreadStarted()563 std::unique_ptr<Thread> prefetch_thread_ TF_GUARDED_BY(*mu_);