Home
last modified time | relevance | path

Searched refs:max_num_threads (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/gemmlowp/eight_bit_int_gemm/
H A Deight_bit_int_gemm.cc257 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 Dmulti_threaded_proxy_resolver.cc119 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 Dmulti_threaded_proxy_resolver.h56 MultiThreadedProxyResolverFactory(size_t max_num_threads,
H A Dmulti_threaded_proxy_resolver_unittest.cc220 size_t max_num_threads, in SingleShotMultiThreadedProxyResolverFactory() argument
222 : MultiThreadedProxyResolverFactory(max_num_threads, false), in SingleShotMultiThreadedProxyResolverFactory()
/aosp_15_r20/external/ruy/ruy/
H A Dtest.h672 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 Dcontext_test.cc33 EXPECT_EQ(context.max_num_threads(), 1); in TEST()
39 EXPECT_EQ(context.max_num_threads(), 2); in TEST()
H A Dcontext.cc44 int Context::max_num_threads() const { return ctx().max_num_threads(); } in max_num_threads() function in ruy::Context
H A Dtrmul.cc264 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 Dcontext.h72 int max_num_threads() const;
H A Dctx.h51 int max_num_threads() const;
H A Dctx.cc45 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 Dmulti_thread_common.h41 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 Dmulti_thread_transform.h36 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareTransform1DTasks()
H A Dmulti_thread_gemm.h60 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareGemmTasks()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dcpu_backend_context.cc141 void CpuBackendContext::SetMaxNumThreads(int max_num_threads) { in SetMaxNumThreads() argument
143 max_num_threads > -1 ? max_num_threads : kDefaultNumThreadpoolThreads; in SetMaxNumThreads()
H A Dcpu_backend_threadpool.h39 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 Dcpu_backend_context.h48 void SetMaxNumThreads(int max_num_threads) override;
50 int max_num_threads() const { return max_num_threads_; } in max_num_threads() function
H A Dadd_n.cc81 cpu_backend_context->max_num_threads()); in Prepare()
H A Dmirror_pad.cc203 const int thread_count = cpu_backend_context->max_num_threads(); in Eval()
/aosp_15_r20/external/gemmlowp/internal/
H A Dmulti_thread_gemm.h546 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 DReduce.cuh103 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 Dexternal_cpu_backend_context.h36 virtual void SetMaxNumThreads(int max_num_threads) = 0;
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dcommon.h1117 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/
Dlib.rs122 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 Ddepthwiseconv_multithread.h135 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConv()

12