Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_state.h57 threadpool_(threadpool), in RPCState()
110 if (threadpool_) { in OnCompleted()
113 threadpool_->Schedule([this]() { ParseAndCallDone(); }); in OnCompleted()
153 thread::ThreadPool* threadpool_; variable
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.cc103 threadpool_(Env::Default(), "GPU_Event_Manager", kNumThreads) { in EventMgr()
104 gpu_event_mgr::InitThreadpoolLabels(&threadpool_); in EventMgr()
136 if (ue->func != nullptr) threadpool_.Schedule(ue->func); in ~EventMgr()
148 threadpool_.Schedule([this]() { PollLoop(); }); in StartPollingLoop()
Dgpu_event_mgr.h145 if (iu.func != nullptr) threadpool_.Schedule(iu.func); in FreeMemory()
202 thread::ThreadPool threadpool_; variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc146 threadpool_(threadpool) { in Dataset()
148 threadpool_->Ref(); in Dataset()
153 threadpool_->Unref(); in ~Dataset()
215 ThreadPoolResource* pool = dataset()->threadpool_; in CreateParams()
229 ThreadPoolResource* const threadpool_; member in tensorflow::data::__anon808a52080111::ThreadPoolDatasetOp::Dataset