Searched refs:underlying_threadpool_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | thread_pool.h | 28 : underlying_threadpool_(tensorflow::Env::Default(), name, num_threads) {} in TfThreadPool()
31 underlying_threadpool_.Schedule(std::move(fn)); in Schedule()
35 underlying_threadpool_.ScheduleWithHint(std::move(fn), start, end); in ScheduleWithHint()
39 underlying_threadpool_.AsEigenThreadPool()->Cancel(); in Cancel()
43 return underlying_threadpool_.NumThreads(); in NumThreads()
47 return underlying_threadpool_.CurrentThreadId(); in CurrentThreadId()
51 tensorflow::thread::ThreadPool underlying_threadpool_;
|
| D | test_util.h | 28 : underlying_threadpool_(tensorflow::Env::Default(), "test_threadpool", in TestThreadPool()
32 underlying_threadpool_.Schedule(std::move(fn)); in Schedule()
36 underlying_threadpool_.ScheduleWithHint(std::move(fn), start, end); in ScheduleWithHint()
42 return underlying_threadpool_.NumThreads(); in NumThreads()
46 return underlying_threadpool_.CurrentThreadId(); in CurrentThreadId()
50 tensorflow::thread::ThreadPool underlying_threadpool_;
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.cc | 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()
245 return underlying_threadpool_->NumThreads(); in NumThreads()
249 return underlying_threadpool_->CurrentThreadId(); in CurrentThreadId()
254 underlying_threadpool_->ScheduleWithHint(std::move(fn), start, limit); in ScheduleWithHint()
268 DCHECK(underlying_threadpool_ != nullptr); in AsEigenThreadPool()
269 return underlying_threadpool_; in AsEigenThreadPool()
|
| D | threadpool.h | 233 Eigen::ThreadPoolInterface* underlying_threadpool_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 47 if (underlying_threadpool_) { in tensorflow_cpu_worker_threads()
74 if (underlying_threadpool_ && underlying_device_->has_eigen_cpu_device()) { in eigen_cpu_device()
84 if (underlying_threadpool_ && in tensorflow_device_thread_pool()
165 std::unique_ptr<thread::ThreadPool> underlying_threadpool_; variable
|
| D | renamed_device.cc | 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()
|
| /external/tensorflow/tensorflow/cc/client/ |
| D | client_session_test.cc | 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::__anonded94f230111::CustomThreadPoolImpl
|