Searched refs:underlying_threadpool_ (Results 1 – 5 of 5) sorted by relevance
103 underlying_threadpool_ = eigen_threadpool_.get(); in ThreadPool()104 threadpool_device_.reset(new Eigen::ThreadPoolDevice(underlying_threadpool_, in ThreadPool()109 underlying_threadpool_ = user_threadpool; in ThreadPool()111 underlying_threadpool_, underlying_threadpool_->NumThreads(), nullptr)); in ThreadPool()118 underlying_threadpool_->Schedule(std::move(fn)); in Schedule()244 return underlying_threadpool_->NumThreads(); in NumThreads()248 return underlying_threadpool_->CurrentThreadId(); in CurrentThreadId()253 underlying_threadpool_->ScheduleWithHint(std::move(fn), start, limit); in ScheduleWithHint()267 DCHECK(underlying_threadpool_ != nullptr); in AsEigenThreadPool()268 return underlying_threadpool_; in AsEigenThreadPool()
232 Eigen::ThreadPoolInterface* underlying_threadpool_; variable
47 if (underlying_threadpool_) { in tensorflow_cpu_worker_threads()73 if (underlying_threadpool_ && underlying_device_->has_eigen_cpu_device()) { in eigen_cpu_device()83 if (underlying_threadpool_ && in tensorflow_device_thread_pool()164 std::unique_ptr<thread::ThreadPool> underlying_threadpool_; variable
60 underlying_threadpool_.reset(new thread::ThreadPool(underlying_threadpool)); in RenamedDevice()61 eigen_worker_threads_.workers = underlying_threadpool_.get(); in RenamedDevice()64 set_tensorflow_device_thread_pool(underlying_threadpool_.get()); in RenamedDevice()
45 underlying_threadpool_.reset(new thread::ThreadPool( in CustomThreadPoolImpl()52 underlying_threadpool_->Schedule(std::move(fn)); in Schedule()57 underlying_threadpool_->ScheduleWithHint(std::move(fn), start, end); in ScheduleWithHint()63 return underlying_threadpool_->NumThreads(); in NumThreads()67 return underlying_threadpool_->CurrentThreadId(); in CurrentThreadId()74 std::unique_ptr<tensorflow::thread::ThreadPool> underlying_threadpool_; member in tensorflow::__anonf29f882d0111::CustomThreadPoolImpl