| /external/pytorch/aten/src/ATen/test/ |
| D | thread_init_test.cpp | 15 ASSERT_EQ(at::get_num_threads(), given_num_threads); in test()
35 ASSERT_TRUE(at::get_num_threads() == 5); in main()
|
| D | test_parallel.cpp | 20 old_num_threads_ = at::get_num_threads(); in NumThreadsGuard()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorIteratorReduce.cpp | 27 if (numel < at::internal::GRAIN_SIZE || at::get_num_threads() == 1 || in parallel_reduce()
42 const int max_threads = at::get_num_threads(); in two_pass_reduction()
77 int num_threads = at::get_num_threads(); in find_split_dim()
138 else if (numel() < at::internal::GRAIN_SIZE || at::get_num_threads() == 1 || in foreach_reduced_elt()
|
| /external/pytorch/aten/src/ATen/ |
| D | ParallelNative.cpp | 133 size_t chunk_size = divup((end - begin), get_num_threads()); in calc_num_tasks_and_chunk_size()
238 int get_num_threads() { in get_num_threads() function
278 if (!in_parallel_region() && get_num_threads() > 1) { in intraop_launch()
295 if (!in_parallel_region() && get_num_threads() > 1) { in intraop_launch_future()
|
| D | Parallel-inl.h | 25 at::get_num_threads() > 1); in parallel_for()
60 const auto max_threads = at::get_num_threads(); in parallel_reduce()
|
| D | ParallelOpenMP.cpp | 77 int get_num_threads() { in get_num_threads() function
|
| D | Parallel.h | 20 TORCH_API int get_num_threads();
|
| D | ParallelCommon.cpp | 55 << at::get_num_threads() << '\n'; in get_parallel_info()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_threadpool.h | 73 virtual int get_num_threads() const override { return num_threads_; } in get_num_threads() function
89 int nthr = get_num_threads(); in parallel_for()
|
| /external/pytorch/test/cpp/api/ |
| D | torch_include.cpp | 12 torch::get_num_threads(); in TEST()
|
| /external/pytorch/torch/csrc/api/include/torch/ |
| D | utils.h | 80 using at::get_num_threads;
|
| /external/cronet/tot/third_party/llvm-libc/src/src/__support/GPU/generic/ |
| D | utils.h | 47 LIBC_INLINE uint64_t get_num_threads() { return 1; } in get_num_threads() function
|
| /external/llvm-libc/test/integration/startup/gpu/ |
| D | rpc_stream_test.cpp | 74 (gpu::get_thread_id() + gpu::get_num_threads() * gpu::get_block_id()) % in test_divergent()
|
| /external/cronet/tot/third_party/llvm-libc/src/test/integration/startup/gpu/ |
| D | rpc_stream_test.cpp | 74 (gpu::get_thread_id() + gpu::get_num_threads() * gpu::get_block_id()) % in test_divergent()
|
| /external/cronet/stable/third_party/llvm-libc/src/src/__support/GPU/generic/ |
| D | utils.h | 47 LIBC_INLINE uint64_t get_num_threads() { return 1; } in get_num_threads() function
|
| /external/cronet/stable/third_party/llvm-libc/src/test/integration/startup/gpu/ |
| D | rpc_stream_test.cpp | 74 (gpu::get_thread_id() + gpu::get_num_threads() * gpu::get_block_id()) % in test_divergent()
|
| /external/cronet/tot/third_party/llvm-libc/src/src/__support/GPU/amdgpu/ |
| D | utils.h | 83 LIBC_INLINE uint64_t get_num_threads() { in get_num_threads() function
|
| /external/cronet/stable/third_party/llvm-libc/src/src/__support/GPU/amdgpu/ |
| D | utils.h | 83 LIBC_INLINE uint64_t get_num_threads() { in get_num_threads() function
|
| /external/llvm-libc/src/__support/GPU/ |
| D | utils.h | 57 LIBC_INLINE uint64_t get_num_threads() { in get_num_threads() function
|
| /external/cronet/stable/third_party/llvm-libc/src/src/__support/GPU/nvptx/ |
| D | utils.h | 77 LIBC_INLINE uint64_t get_num_threads() { in get_num_threads() function
|
| /external/cronet/tot/third_party/llvm-libc/src/src/__support/GPU/nvptx/ |
| D | utils.h | 77 LIBC_INLINE uint64_t get_num_threads() { in get_num_threads() function
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | SerialStackImpl.h | 117 return numel_in_stack < at::internal::GRAIN_SIZE || at::get_num_threads() == 1; in can_use_native_serial_stack_impl()
|
| D | Reduce.h | 229 if (numel < at::internal::GRAIN_SIZE || at::get_num_threads() == 1 || in binary_kernel_reduce()
233 int max_threads = at::get_num_threads(); in binary_kernel_reduce()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qlinear_dynamic.cpp | 126 int num_tasks = at::get_num_threads(); in apply_dynamic_impl()
432 int num_tasks = at::get_num_threads(); in apply_dynamic_impl()
576 int num_threads = at::get_num_threads(); in apply_dynamic_impl()
|
| /external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
| D | qlinear.cpp | 136 int num_tasks = at::get_num_threads(); in apply_impl()
|