| /external/pytorch/caffe2/utils/threadpool/ |
| D | pthreadpool-cpp.cc | 23 : threadpool_(pthreadpool_create(thread_count), pthreadpool_destroy) {} in PThreadPool()
28 TORCH_INTERNAL_ASSERT(threadpool_.get(), "Invalid threadpool!"); in get_thread_count()
29 return pthreadpool_get_threads_count(threadpool_.get()); in get_thread_count()
46 threadpool_.reset(pthreadpool_create(thread_count)); in set_thread_count()
63 TORCH_INTERNAL_ASSERT(threadpool_.get(), "Invalid threadpool!"); in run()
72 threadpool_.get(), in run()
115 return threadpool->threadpool_.get(); in pthreadpool_()
|
| D | pthreadpool-cpp.h | 40 std::unique_ptr<pthreadpool, decltype(&pthreadpool_destroy)> threadpool_; variable
|
| /external/executorch/extension/threadpool/ |
| D | threadpool.cpp | 38 : threadpool_(pthreadpool_create(thread_count), pthreadpool_destroy) {} in ThreadPool()
43 ET_CHECK_MSG(threadpool_.get(), "Invalid threadpool!"); in get_thread_count()
44 return pthreadpool_get_threads_count(threadpool_.get()); in get_thread_count()
55 threadpool_.reset(pthreadpool_create(new_thread_count)); in _unsafe_reset_threadpool()
73 ET_CHECK_MSG(threadpool_.get(), "Invalid threadpool!"); in run()
82 threadpool_.get(), in run()
138 return threadpool->threadpool_.get(); in get_pthreadpool()
|
| D | threadpool.h | 67 std::unique_ptr<pthreadpool, decltype(&pthreadpool_destroy)> threadpool_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.cc | 98 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()
|
| D | device_event_mgr.h | 104 if (iu.func != nullptr) threadpool_.Schedule(iu.func); in FreeMemory()
145 thread::ThreadPool threadpool_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr.cc | 97 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/grappler/optimizers/ |
| D | tfg_optimizer_hook.cc | 70 threadpool_ = std::make_unique<llvm::ThreadPool>(strategy); in Impl()
71 ctx_.setThreadPool(*threadpool_); in Impl()
92 std::unique_ptr<llvm::ThreadPool> threadpool_; member in mlir::tfg::TFGGrapplerOptimizer::Impl
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.h | 33 thread::ThreadPool* GetThreadPool() const { return threadpool_.get(); } in GetThreadPool()
59 std::unique_ptr<thread::ThreadPool> threadpool_; variable
|
| D | grpc_state.h | 92 threadpool_(threadpool), in RPCState()
149 if (threadpool_) { in OnCompleted()
152 threadpool_->Schedule([this]() { ParseAndCallDone(); }); in OnCompleted()
218 thread::ThreadPool* threadpool_; variable
|
| D | grpc_worker_cache.cc | 142 : threadpool_(new thread::ThreadPool( in GrpcWorkerEnv()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 172 threadpool_(threadpool) { in Dataset()
174 threadpool_->Ref(); in Dataset()
179 threadpool_->Unref(); in ~Dataset()
266 ThreadPoolResource* pool = dataset()->threadpool_; in CreateParams()
280 ThreadPoolResource* const threadpool_; member in tensorflow::data::experimental::ThreadPoolDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 164 threadpool_.reset( in Delegate()
204 return threadpool_.get(); in threadpool()
243 std::unique_ptr<pthreadpool, decltype(&pthreadpool_destroy)> threadpool_{
|