Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/binaries/
H A Dat_launch_benchmark.cc69 << at::get_num_interop_threads() << " threads " in main()
80 << at::get_num_interop_threads() << " threads " in main()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtorch_include.cpp13 torch::get_num_interop_threads(); in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dthread_init_test.cpp40 ASSERT_EQ(at::get_num_interop_threads(), 5); in main()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dutils.h86 using at::get_num_interop_threads;
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DParallelThreadPoolNative.cpp59 int get_num_interop_threads() { in get_num_interop_threads() function
H A DParallel.h136 TORCH_API int get_num_interop_threads();
H A DParallelCommon.cpp57 << at::get_num_interop_threads() << '\n'; in get_parallel_info()
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dcpu_threading_torchscript_inference.rst97 | | ``at::get_num_interop_threads`` (C++) | …
100 | | ``get_num_interop_threads`` (Python, :mod:`torch` module) | …
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.rst255 get_num_interop_threads
/aosp_15_r20/external/pytorch/torch/
H A D_torch_docs.py4892 torch.get_num_interop_threads,
H A Doverrides.py134 torch.get_num_interop_threads,
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp327 return THPUtils_packInt32(at::get_num_interop_threads()); in THPModule_getNumInteropThreads()
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1131 def get_num_interop_threads() -> _int: ... # THPModule_getNumInteropThreads