Home
last modified time | relevance | path

Searched refs:threadpool_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.cc97 threadpool_(Env::Default(), "Device_Event_Manager", kNumThreads) { in EventMgr()
98 device_event_mgr::InitThreadpoolLabels(&threadpool_); in EventMgr()
112 if (ue->func != nullptr) threadpool_.Schedule(ue->func); in ~EventMgr()
124 threadpool_.Schedule([this]() { PollLoop(); }); in StartPollingLoop()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr.cc98 threadpool_(Env::Default(), "Device_Event_Manager", kNumThreads) { in EventMgr()
99 device_event_mgr::InitThreadpoolLabels(&threadpool_); in EventMgr()
113 if (ue->func != nullptr) threadpool_.Schedule(ue->func); in ~EventMgr()
125 threadpool_.Schedule([this]() { PollLoop(); }); in StartPollingLoop()
Ddevice_event_mgr.h104 if (iu.func != nullptr) threadpool_.Schedule(iu.func); in FreeMemory()
145 thread::ThreadPool threadpool_; variable
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache.h33 thread::ThreadPool* GetThreadPool() const { return threadpool_.get(); } in GetThreadPool()
59 std::unique_ptr<thread::ThreadPool> threadpool_; variable
Dgrpc_state.h98 threadpool_(threadpool), in RPCState()
155 if (threadpool_) { in OnCompleted()
158 threadpool_->Schedule([this]() { ParseAndCallDone(); }); in OnCompleted()
209 thread::ThreadPool* threadpool_; variable
Dgrpc_worker_cache.cc122 : threadpool_(new thread::ThreadPool( in GrpcWorkerEnv()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc147 threadpool_(threadpool) { in Dataset()
149 threadpool_->Ref(); in Dataset()
154 threadpool_->Unref(); in ~Dataset()
239 ThreadPoolResource* pool = dataset()->threadpool_; in CreateParams()
253 ThreadPoolResource* const threadpool_; member in tensorflow::data::experimental::__anon3c251a260111::ThreadPoolDatasetOp::Dataset
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc52 threadpool_.reset( in Delegate()
67 return threadpool_.get(); in threadpool()
96 std::unique_ptr<pthreadpool, decltype(&pthreadpool_destroy)> threadpool_{