| /aosp_15_r20/system/core/init/ |
| H A D | ueventd_test.cpp | 41 auto num_threads = files_and_parameters.size(); in WriteFromMultipleThreads() local
131 constexpr auto num_threads = 10; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_set_diag_op_gpu.cu.cc | 47 const int num_threads, const int m, const int n, const int num_diags, in MatrixSetDiagKernel()
73 const int num_threads, const int m, const int n, const int num_diags, in MatrixCopyInputAndSetDiagKernel()
|
| H A D | matrix_diag_op_gpu.cu.cc | 46 const int num_threads, const int num_rows, const int num_cols, in MatrixDiagKernel()
103 const int num_threads, const int num_rows, const int num_cols, in MatrixDiagPartKernel()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| H A D | tflite_engine.cc | 194 absl::Status TfLiteEngine::InitInterpreter(int num_threads) { in InitInterpreter()
216 const tflite::proto::ComputeSettings& compute_settings, int num_threads) { in InitInterpreter()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_worker_cache.cc | 141 GrpcWorkerEnv::GrpcWorkerEnv(size_t num_completion_queues, size_t num_threads) in GrpcWorkerEnv()
174 int64_t num_threads; in CreateGrpcWorkerEnv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ScanUtils.cuh | 63 … const uint32_t num_threads, const uint32_t log_num_threads_x, in tensor_kernel_scan_innermost_dim_with_indices()
226 const uint32_t num_threads = 512; in scan_innermost_dim_with_indices() local
425 const uint32_t num_threads = 512; in scan_innermost_dim() local
|
| H A D | RangeFactories.cu | 27 constexpr int num_threads() { in num_threads() function
39 C10_LAUNCH_BOUNDS_1(num_threads()) in C10_LAUNCH_BOUNDS_1() argument
|
| H A D | UpSampleBilinear2d.cu | 307 const int num_threads = std::min( in upsample_bilinear2d_out_cuda_template() local
334 const int num_threads = std::min( in upsample_bilinear2d_out_cuda_template() local
391 const int num_threads = std::min( in upsample_bilinear2d_backward_out_cuda_template() local
794 const int num_threads = std::min(at::cuda::getCurrentDeviceProperties()->maxThreadsPerBlock, 256); in upsample_gen2d_aa_backward_out_cuda_template() local
|
| H A D | UpSampleLinear1d.cu | 137 const int num_threads = 512; in upsample_linear1d_out_cuda_template() local
181 const int num_threads = 512; in upsample_linear1d_backward_out_cuda_template() local
|
| H A D | CUDALoops.cuh | 57 C10_LAUNCH_BOUNDS_1(num_threads()) in C10_LAUNCH_BOUNDS_1() argument
90 C10_LAUNCH_BOUNDS_1(num_threads()) in C10_LAUNCH_BOUNDS_1() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/ |
| H A D | run_handler_test.cc | 43 int num_threads = 2; in TEST() local
84 int num_threads = 2; in TEST() local
121 int num_threads = 2; in TEST() local
|
| /aosp_15_r20/external/google-breakpad/src/common/linux/tests/ |
| H A D | crash_generator.cc | 185 unsigned num_threads, unsigned crash_thread, int crash_signal, in CreateChildCrash()
294 void CrashGenerator::CreateThreadsInChildProcess(unsigned num_threads) { in CreateThreadsInChildProcess()
|
| /aosp_15_r20/art/runtime/ |
| H A D | thread_pool_test.cc | 56 static int32_t num_threads; member in art::ThreadPoolTest
59 int32_t ThreadPoolTest::num_threads = 4; member in art::ThreadPoolTest
|
| /aosp_15_r20/external/rust/android-crates-io/crates/rayon-core/src/ |
| D | lib.rs | 182 num_threads: usize, field
539 pub fn num_threads(mut self, num_threads: usize) -> Self { in num_threads() method
701 pub fn num_threads(mut self, num_threads: usize) -> Configuration { in num_threads() method
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
| H A D | lp_query.c | 99 const unsigned num_threads = MAX2(1, screen->num_threads); in llvmpipe_get_query_result() local
226 const unsigned num_threads = MAX2(1, screen->num_threads); in llvmpipe_get_query_result_resource() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 884 Transformation transformation, int num_threads) { in Create()
1192 int num_threads = in Initialize() local
1336 int num_threads; member
1370 TransposePlan::Transformation transformation, int num_threads) { in GetOrCreate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | redux_functor.h | 66 const Eigen::Index num_threads = device.numThreads(); in operator() local
247 const Eigen::Index num_threads = device.numThreads(); in operator() local
|
| /aosp_15_r20/external/cronet/base/allocator/partition_allocator/src/partition_alloc/ |
| H A D | partition_lock_unittest.cc | 71 constexpr int num_threads = 4; in TEST() local
135 constexpr int num_threads = 4; in TEST() local
|
| /aosp_15_r20/external/igt-gpu-tools/tests/ |
| H A D | prime_self_import.c | 238 int r, i, num_threads; in test_reimport_close_race() local
325 int r, i, num_threads; in test_export_close_race() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | FusedObsFakeQuant.cu | 154 int num_threads = std::min(size, (int64_t)512); in _calculate_moving_average() local
204 int num_threads = std::min(size, (int64_t)512); in _calc_moving_avg_qparams_helper() local
|
| /aosp_15_r20/external/cronet/base/threading/ |
| H A D | thread_local_storage_perftest.cc | 116 size_t num_threads) { in Benchmark()
146 size_t num_threads) { in BenchmarkImpl()
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | weights-cache.cc | 241 constexpr size_t num_threads = 20; in TEST() local
274 constexpr size_t num_threads = 20; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/client/ |
| H A D | client_session_test.cc | 143 int num_threads = 3; in TEST() local
207 int num_threads = 3; in TEST() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/threadpool/src/ |
| D | lib.rs | 159 num_threads: Option<usize>, field
208 pub fn num_threads(mut self, num_threads: usize) -> Builder { in num_threads() method
284 let num_threads = self.num_threads.unwrap_or_else(num_cpus::get); in build() localVariable
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_micro_gemm.py | 470 def check_amx_extra(config, m, n, k, alpha, num_threads): argument
754 num_threads=-1, argument
|