Caffe2 - C++ API
A deep learning, cross platform ML framework
net_async_gpu_thread_pool.h
1 
17 #ifndef CAFFE2_CORE_NET_ASYNC_GPU_THREAD_POOL_H_
18 #define CAFFE2_CORE_NET_ASYNC_GPU_THREAD_POOL_H_
19 
20 #include "caffe2/core/net_async_polling.h"
21 
22 namespace caffe2 {
23 
24 std::shared_ptr<TaskThreadPool> GetAsyncNetGPUThreadPool(int gpu_id);
25 
26 } // namespace caffe2
27 
28 #endif // CAFFE2_CORE_NET_ASYNC_GPU_THREAD_POOL_H_
Copyright (c) 2016-present, Facebook, Inc.