Lines Matching refs:TORCH_API

14 TORCH_API void init_num_threads();
17 TORCH_API void set_num_threads(int);
20 TORCH_API int get_num_threads();
24 TORCH_API int get_thread_num();
27 TORCH_API bool in_parallel_region();
40 TORCH_API void set_thread_num(int);
42 class TORCH_API ThreadIdGuard {
130 TORCH_API std::string get_parallel_info();
133 TORCH_API void set_num_interop_threads(int);
136 TORCH_API int get_num_interop_threads();
139 TORCH_API void launch(std::function<void()> func);
145 TORCH_API void intraop_launch(std::function<void()> func);
148 TORCH_API int intraop_default_num_threads();