Searched refs:PrefetchThread (Results 1 – 3 of 3) sorted by relevance
239 PrefetchThread::PrefetchThread(std::unique_ptr<TaskIterator> iterator, in PrefetchThread() function in tensorflow::data::PrefetchThread246 PrefetchThread::~PrefetchThread() { in ~PrefetchThread()252 void PrefetchThread::Run() { in Run()288 Status PrefetchThread::FillBuffer(int64 wait_us, in FillBuffer()316 Status PrefetchThread::GetStatus() { in GetStatus()
82 class PrefetchThread {84 explicit PrefetchThread(std::unique_ptr<TaskIterator> iterator,86 ~PrefetchThread();166 PrefetchThread prefetch_thread_;
436 "tf_data_prefetch", [this, new_ctx]() { PrefetchThread(new_ctx); }); in EnsurePrefetchThreadStarted()444 void PrefetchThread(const std::shared_ptr<IteratorContext>& ctx) { in PrefetchThread() function in tensorflow::data::PrefetchDatasetOp::Dataset::Iterator