/aosp_15_r20/external/gemmlowp/eight_bit_int_gemm/ |
H A D | eight_bit_int_gemm.cc | 257 const std::int32_t max_num_threads = context->max_num_threads(); in MetaGemmQuantized8Bit() local 259 scratch->AssureSize(meta::gemm_q8_scratch(m, n, k, max_num_threads)); in MetaGemmQuantized8Bit() 260 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit() 265 scratch->AssureSize(meta::gemm_q8_scratch(n, m, k, max_num_threads)); in MetaGemmQuantized8Bit() 266 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit() 281 const std::int32_t max_num_threads = context->max_num_threads(); in MetaGemmFloat() local 283 scratch->AssureSize(meta::gemm_f_scratch(m, n, k, max_num_threads)); in MetaGemmFloat() 284 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat() 288 scratch->AssureSize(meta::gemm_f_scratch(n, m, k, max_num_threads)); in MetaGemmFloat() 289 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat()
|
/aosp_15_r20/external/cronet/net/proxy_resolution/ |
H A D | multi_threaded_proxy_resolver.cc | 119 size_t max_num_threads, 424 size_t max_num_threads, in MultiThreadedProxyResolver() argument 428 max_num_threads_(max_num_threads), in MultiThreadedProxyResolver() 527 size_t max_num_threads, in Job() argument 532 max_num_threads_(max_num_threads), in Job() 580 size_t max_num_threads, in MultiThreadedProxyResolverFactory() argument 583 max_num_threads_(max_num_threads) { in MultiThreadedProxyResolverFactory() 584 DCHECK_GE(max_num_threads, 1u); in MultiThreadedProxyResolverFactory()
|
H A D | multi_threaded_proxy_resolver.h | 56 MultiThreadedProxyResolverFactory(size_t max_num_threads,
|
H A D | multi_threaded_proxy_resolver_unittest.cc | 220 size_t max_num_threads, in SingleShotMultiThreadedProxyResolverFactory() argument 222 : MultiThreadedProxyResolverFactory(max_num_threads, false), in SingleShotMultiThreadedProxyResolverFactory()
|
/aosp_15_r20/external/ruy/ruy/ |
H A D | test.h | 672 int max_num_threads = 0; 724 if (max_num_threads) { 725 GlobalContext().set_max_num_threads(max_num_threads); 788 const MulParamsType& mul_params, int max_num_threads, 828 GlobalGemmlowpContext().set_max_num_threads(max_num_threads ? max_num_threads 887 const MulParamsType& mul_params, int max_num_threads, 894 return EvalGemmlowp<LHS, RHS, DST>(lhs, rhs, mul_params, max_num_threads, \ 931 const MulParamsType& mul_params, int max_num_threads, 964 Eigen::setNbThreads(max_num_threads ? max_num_threads : 1); 991 const MulParamsType& mul_params, int max_num_threads, [all …]
|
H A D | context_test.cc | 33 EXPECT_EQ(context.max_num_threads(), 1); in TEST() 39 EXPECT_EQ(context.max_num_threads(), 2); in TEST()
|
H A D | context.cc | 44 int Context::max_num_threads() const { return ctx().max_num_threads(); } in max_num_threads() function in ruy::Context
|
H A D | trmul.cc | 264 return ctx->max_num_threads(); in GetTentativeThreadCount() 279 std::min<std::int64_t>(tentative_thread_count, ctx->max_num_threads()); in GetTentativeThreadCount() 313 static_cast<int>(params->path), ctx->max_num_threads(), in TrMul()
|
H A D | context.h | 72 int max_num_threads() const;
|
H A D | ctx.h | 51 int max_num_threads() const;
|
H A D | ctx.cc | 45 int Ctx::max_num_threads() const { return impl().max_num_threads_; } in max_num_threads() function in ruy::Ctx
|
/aosp_15_r20/external/gemmlowp/meta/ |
H A D | multi_thread_common.h | 41 SimpleContext(int max_num_threads, WorkersPool* pool) in SimpleContext() argument 42 : max_num_threads_(max_num_threads), pool_(pool) {} in SimpleContext() 46 int max_num_threads() { return max_num_threads_; } in max_num_threads() function
|
H A D | multi_thread_transform.h | 36 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareTransform1DTasks()
|
H A D | multi_thread_gemm.h | 60 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareGemmTasks()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | cpu_backend_context.cc | 141 void CpuBackendContext::SetMaxNumThreads(int max_num_threads) { in SetMaxNumThreads() argument 143 max_num_threads > -1 ? max_num_threads : kDefaultNumThreadpoolThreads; in SetMaxNumThreads()
|
H A D | cpu_backend_threadpool.h | 39 TFLITE_DCHECK_LE(tasks_count, cpu_backend_context->max_num_threads()); in Execute() 51 TFLITE_DCHECK_LE(tasks_count, cpu_backend_context->max_num_threads());
|
H A D | cpu_backend_context.h | 48 void SetMaxNumThreads(int max_num_threads) override; 50 int max_num_threads() const { return max_num_threads_; } in max_num_threads() function
|
H A D | add_n.cc | 81 cpu_backend_context->max_num_threads()); in Prepare()
|
H A D | mirror_pad.cc | 203 const int thread_count = cpu_backend_context->max_num_threads(); in Eval()
|
/aosp_15_r20/external/gemmlowp/internal/ |
H A D | multi_thread_gemm.h | 546 int max_num_threads() const { return max_num_threads_; } in max_num_threads() function 580 inline int HowManyThreads(int max_num_threads, int rows, int cols, int depth) { in HowManyThreads() argument 582 if (max_num_threads == 1) { in HowManyThreads() 587 int max_count = GetHardwareConcurrency(max_num_threads); in HowManyThreads() 659 context->max_num_threads(), rows, cols, depth); in MultiThreadGemm()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | Reduce.cuh | 103 const int max_num_threads = mnt_wrapper<T>::MAX_NUM_THREADS / output_vec_size; in set_block_dimension() local 104 int dim0_pow2 = dim0 < max_num_threads ? static_cast<int>(last_pow2(dim0)) : max_num_threads; in set_block_dimension() 105 int dim1_pow2 = dim1 < max_num_threads ? static_cast<int>(last_pow2(dim1)) : max_num_threads; in set_block_dimension() 107 block_height = std::min(dim1_pow2, int(max_num_threads / block_width)); in set_block_dimension() 108 block_width = std::min(dim0_pow2, int(max_num_threads / block_height)); in set_block_dimension()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/ |
H A D | external_cpu_backend_context.h | 36 virtual void SetMaxNumThreads(int max_num_threads) = 0;
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
H A D | common.h | 1117 inline int LegacyHowManyThreads(int max_num_threads, int rows, int cols, in LegacyHowManyThreads() argument 1120 if (max_num_threads == 1) { in LegacyHowManyThreads() 1125 int thread_count = std::min(max_num_threads, rows / KernelRows); in LegacyHowManyThreads() 1144 assert(thread_count > 0 && thread_count <= max_num_threads); in LegacyHowManyThreads()
|
/aosp_15_r20/external/rust/android-crates-io/crates/rayon/src/ |
D | lib.rs | 122 pub use rayon_core::{current_num_threads, current_thread_index, max_num_threads};
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | depthwiseconv_multithread.h | 135 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConv()
|