Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DParallel-inl.h21 at::internal::lazy_init_num_threads(); in parallel_for()
59 at::internal::lazy_init_num_threads(); in parallel_reduce()
H A DParallelNative.cpp145 at::internal::lazy_init_num_threads(); in invoke_parallel()
239 at::internal::lazy_init_num_threads(); in get_num_threads()
H A DParallelOpenMP.cpp79 at::internal::lazy_init_num_threads(); in get_num_threads()
H A DParallelThreadPoolNative.cpp60 at::internal::lazy_init_num_threads(); in get_num_interop_threads()
H A DParallel.h32 inline void lazy_init_num_threads() { in lazy_init_num_threads() function
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1644 at::internal::lazy_init_num_threads(); in initModule()