Caffe2 - C++ API
A deep learning, cross platform ML framework
Data Structures | Public Member Functions
caffe2::TaskThreadPool Class Reference

Public Member Functions

 TaskThreadPool (std::size_t pool_size, int numa_node_id=-1)
 
template<typename Task >
void runTask (Task task)
 Add task to the thread pool if a thread is currently available.
 
void run (const std::function< void()> &func)
 
template<typename Task >
void runTaskWithID (Task task)
 
void waitWorkComplete ()
 Wait for queue to be empty.
 

Detailed Description

Definition at line 15 of file thread_pool.h.


The documentation for this class was generated from the following file: