Home
last modified time | relevance | path

Searched refs:thread_pool_shared_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp622 auto tp_shared = thread_pool_shared_; in reentrant_thread_init()
625 ++thread_pool_shared_->num_workers_; in reentrant_thread_init()
628 --thread_pool_shared_->num_workers_; in reentrant_thread_init()
1476 thread_pool_shared_ = std::make_shared<ThreadPoolShared>(); in start_device_threads()
1521 std::unique_lock<std::mutex> lck(thread_pool_shared_->mutex_); in add_thread_pool_task()
1526 (thread_pool_shared_->num_workers_ <= in add_thread_pool_task()
1527 thread_pool_shared_->graphtasks_queue_.size()); in add_thread_pool_task()
1528 thread_pool_shared_->graphtasks_queue_.push(graph_task); in add_thread_pool_task()
1539 thread_pool_shared_->work_.notify_one(); in add_thread_pool_task()
H A Dengine.h269 std::shared_ptr<ThreadPoolShared> thread_pool_shared_; member