| /external/pytorch/aten/src/ATen/test/ |
| D | test_parallel.cpp | 21 at::set_num_threads(nthreads); in NumThreadsGuard()
25 at::set_num_threads(old_num_threads_); in ~NumThreadsGuard()
61 set_num_threads(nthreads); in TEST()
|
| D | thread_init_test.cpp | 25 at::set_num_threads(4); in main()
34 at::set_num_threads(5); in main()
|
| /external/pytorch/aten/src/ATen/ |
| D | ParallelOpenMP.cpp | 32 set_num_threads(nthreads); in init_num_threads()
46 void set_num_threads(int nthreads) { in set_num_threads() function
|
| D | Parallel.h | 17 TORCH_API void set_num_threads(int);
|
| /external/pytorch/test/ |
| D | test_openmp.py | 62 torch.set_num_threads(1)
68 torch.set_num_threads(ncores)
|
| /external/pytorch/docs/source/notes/ |
| D | cpu_threading_torchscript_inference.rst | 102 | Intra-op parallelism | ``at::set_num_threads``, | …
104 | | ``set_num_threads``, | …
111 For the intra-op parallelism settings, ``at::set_num_threads``, ``torch.set_num_threads`` always ta…
125 torch.set_num_threads(t)
138 … find disabling intra-op parallelism as a possible option (i.e. by calling ``set_num_threads(1)``);
|
| /external/armnn/src/backends/neon/ |
| D | NeonInterceptorScheduler.cpp | 15 void NeonInterceptorScheduler::set_num_threads(unsigned int numThreads) in set_num_threads() function in armnn::NeonInterceptorScheduler
17 m_RealScheduler.set_num_threads(numThreads); in set_num_threads()
|
| D | NeonInterceptorScheduler.hpp | 23 void set_num_threads(unsigned int numThreads) override;
|
| /external/pytorch/test/cpp/api/ |
| D | torch_include.cpp | 10 torch::set_num_threads(2); in TEST()
|
| /external/rust/android-crates-io/crates/threadpool/src/ |
| D | lib.rs | 527 self.set_num_threads(num_threads) in set_threads()
570 pub fn set_num_threads(&mut self, num_threads: usize) { in set_num_threads() method
801 pool.set_num_threads(new_thread_amount); in test_set_num_threads_increasing()
821 pool.set_num_threads(new_thread_amount); in test_set_num_threads_decreasing()
980 pool.set_num_threads(TEST_TASKS); in test_shrink()
1006 pool.set_num_threads(3); in test_name()
|
| /external/ComputeLibrary/arm_compute/runtime/ |
| D | SingleThreadScheduler.h | 41 void set_num_threads(unsigned int num_threads) override;
|
| D | IScheduler.h | 146 virtual void set_num_threads(unsigned int num_threads) = 0;
|
| /external/ComputeLibrary/arm_compute/runtime/OMP/ |
| D | OMPScheduler.h | 41 void set_num_threads(unsigned int num_threads) override;
|
| /external/ComputeLibrary/src/runtime/CPP/ |
| D | CPPScheduler.cpp | 342 void set_num_threads(unsigned int num_threads, unsigned int thread_hint) in set_num_threads() function
445 void CPPScheduler::set_num_threads(unsigned int num_threads) in set_num_threads() function in arm_compute::CPPScheduler
449 _impl->set_num_threads(num_threads, num_threads_hint()); in set_num_threads()
|
| D | SingleThreadScheduler.cpp | 32 void SingleThreadScheduler::set_num_threads(unsigned int num_threads) in set_num_threads() function in arm_compute::SingleThreadScheduler
|
| /external/ComputeLibrary/arm_compute/runtime/CPP/ |
| D | CPPScheduler.h | 58 void set_num_threads(unsigned int num_threads) override;
|
| /external/pytorch/torch/utils/benchmark/utils/valgrind_wrapper/ |
| D | timer_callgrind_template.cpp | 43 torch::set_num_threads(number_threads); in main()
|
| /external/pytorch/torch/csrc/api/include/torch/ |
| D | utils.h | 83 using at::set_num_threads;
|
| /external/libopus/dnn/torch/lpcnet/ |
| D | test_lpcnet.py | 60 torch.set_num_threads(2)
|
| /external/libopus/dnn/torch/osce/ |
| D | test_model.py | 61 torch.set_num_threads(2)
|
| D | test_vocoder.py | 63 torch.set_num_threads(2)
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/ |
| D | evaluation_delegate_provider_test.cc | 68 params.set_num_threads(4); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | inference_profiler_stage_test.cc | 41 params->set_num_threads(num_threads); in GetInferenceProfilerStageConfig()
|
| /external/pytorch/benchmarks/operator_benchmark/ |
| D | benchmark_runner.py | 159 torch.set_num_threads(args.omp_num_threads)
|
| /external/ComputeLibrary/tests/framework/instruments/ |
| D | SchedulerTimer.cpp | 64 void set_num_threads(unsigned int num_threads) override in set_num_threads() function in arm_compute::test::framework::Interceptor
66 _real_scheduler.set_num_threads(num_threads); in set_num_threads()
|