| /external/gemmlowp/eight_bit_int_gemm/ |
| 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()
|
| /external/gemmlowp/meta/ |
| 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
|
| D | multi_thread_transform.h | 36 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareTransform1DTasks()
|
| D | multi_thread_gemm.h | 60 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareGemmTasks()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | cpu_backend_context.cc | 140 void CpuBackendContext::SetMaxNumThreads(int max_num_threads) { in SetMaxNumThreads() argument
142 max_num_threads > -1 ? max_num_threads : kDefaultNumThreadpoolThreads; in SetMaxNumThreads()
|
| 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());
|
| 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
|
| D | mirror_pad.cc | 203 const int thread_count = cpu_backend_context->max_num_threads(); in Eval()
|
| D | cpu_backend_gemm_custom_gemv.h | 165 context->max_num_threads(), dst_params.rows, dst_params.cols, in CustomGemv()
|
| D | fully_connected.cc | 600 const int max_threads = cpu_backend_context->max_num_threads(); in EvalHybrid()
|
| /external/ruy/ruy/ |
| D | context_test.cc | 32 EXPECT_EQ(context.max_num_threads(), 1); in TEST()
36 EXPECT_EQ(context.max_num_threads(), 2); in TEST()
|
| D | context.cc | 43 int Context::max_num_threads() const { return ctx().max_num_threads(); } in max_num_threads() function in ruy::Context
|
| D | test.h | 671 int max_num_threads = 0;
723 if (max_num_threads) {
724 GlobalContext().set_max_num_threads(max_num_threads);
787 const MulParamsType& mul_params, int max_num_threads,
827 GlobalGemmlowpContext().set_max_num_threads(max_num_threads ? max_num_threads
886 const MulParamsType& mul_params, int max_num_threads,
893 return EvalGemmlowp<LHS, RHS, DST>(lhs, rhs, mul_params, max_num_threads, \
930 const MulParamsType& mul_params, int max_num_threads,
963 Eigen::setNbThreads(max_num_threads ? max_num_threads : 1);
990 const MulParamsType& mul_params, int max_num_threads,
[all …]
|
| D | context.h | 71 int max_num_threads() const;
|
| D | ctx.h | 50 int max_num_threads() const;
|
| D | trmul.cc | 262 std::min(1 << guess_log2, ctx->max_num_threads()); in GetTentativeThreadCount()
294 static_cast<int>(params->path), ctx->max_num_threads(), in TrMul()
|
| D | ctx.cc | 44 int Ctx::max_num_threads() const { return impl().max_num_threads_; } in max_num_threads() function in ruy::Ctx
|
| D | trace.h | 606 tentative_thread_count, rows, depth, cols, ctx->max_num_threads());
|
| /external/gemmlowp/internal/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/ |
| D | external_cpu_backend_context.h | 36 virtual void SetMaxNumThreads(int max_num_threads) = 0;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_multithread.h | 133 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 974 inline int LegacyHowManyThreads(int max_num_threads, int rows, int cols, in LegacyHowManyThreads() argument
977 if (max_num_threads == 1) { in LegacyHowManyThreads()
982 int thread_count = std::min(max_num_threads, rows / KernelRows); in LegacyHowManyThreads()
1001 assert(thread_count > 0 && thread_count <= max_num_threads); in LegacyHowManyThreads()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 174 const int max_threads = cpu_backend_context->max_num_threads(); in FullyConnectedSparseWeight1x4()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | mean.h | 219 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
|
| D | depthwise_conv_hybrid.h | 420 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConvHybridPerChannel()
|