Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/
DParallelNative.cpp76 TaskThreadPoolBase& _get_intraop_pool() { in _get_intraop_pool() function
93 _get_intraop_pool().run([fn, i]() { fn((int)i, i); }); in _run_with_pool()
222 stored_nthreads = _get_intraop_pool().size() + 1; in set_num_threads()
251 return _get_intraop_pool().size() + 1; in get_num_threads()
269 _get_intraop_pool().inThreadPool() in in_parallel_region()
279 _get_intraop_pool().run(std::move(func)); in intraop_launch()
296 _get_intraop_pool().run( in intraop_launch_future()