Home
last modified time | relevance | path

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

/external/pytorch/caffe2/utils/threadpool/
Dpthreadpool-cpp.h17 class PThreadPool final {
19 explicit PThreadPool(size_t thread_count);
20 ~PThreadPool() = default;
22 PThreadPool(const PThreadPool&) = delete;
23 PThreadPool& operator=(const PThreadPool&) = delete;
25 PThreadPool(PThreadPool&&) = delete;
26 PThreadPool& operator=(PThreadPool&&) = delete;
44 PThreadPool* pthreadpool();
Dpthreadpool-cpp.cc22 PThreadPool::PThreadPool(const size_t thread_count) in PThreadPool() function in caffe2::PThreadPool
25 size_t PThreadPool::get_thread_count() const { in get_thread_count()
32 void PThreadPool::set_thread_count(const size_t thread_count) { in set_thread_count()
49 void PThreadPool::run( in run()
88 PThreadPool* pthreadpool() { in pthreadpool()
90 std::make_unique<PThreadPool>(getDefaultNumThreads()); in pthreadpool()
102 threadpool.reset(new PThreadPool(num_threads)); in pthreadpool()
112 PThreadPool* const threadpool = pthreadpool(); in pthreadpool_()
/external/pytorch/aten/src/ATen/
DParallelNative.cpp98 caffe2::PThreadPool* const pool = caffe2::pthreadpool(); in _run_with_pool()
232 caffe2::PThreadPool* const pool = caffe2::pthreadpool(); in set_num_threads()
254 caffe2::PThreadPool* const pool = caffe2::pthreadpool(); in get_num_threads()
DParallelOpenMP.cpp64 caffe2::PThreadPool* const pool = caffe2::pthreadpool(); in set_num_threads()