| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/ruy/ruy/ |
| A D | context_test.cc | 32 EXPECT_EQ(context.max_num_threads(), 1); in TEST()
36 EXPECT_EQ(context.max_num_threads(), 2); in TEST()
|
| A D | context.cc | 43 int Context::max_num_threads() const { return ctx().max_num_threads(); } in max_num_threads() function in ruy::Context
|
| A D | test.h | 671 int max_num_threads = 0;
723 if (max_num_threads) {
724 GlobalContext().set_max_num_threads(max_num_threads);
827 GlobalGemmlowpContext().set_max_num_threads(max_num_threads ? max_num_threads
963 Eigen::setNbThreads(max_num_threads ? max_num_threads : 1);
1070 static Eigen::ThreadPool pool(max_num_threads ? max_num_threads : 1);
1112 max_num_threads, dst);
1243 Eigen::setNbThreads(max_num_threads ? max_num_threads : 1);
1802 if (max_num_threads == 0) {
1803 max_num_threads = GetIntEnvVarOrZero("THREADS");
[all …]
|
| A D | context.h | 71 int max_num_threads() const;
|
| A D | ctx.h | 50 int max_num_threads() const;
|
| A D | trmul.cc | 263 std::min(1 << guess_log2, ctx->max_num_threads()); in GetTentativeThreadCount()
295 static_cast<int>(params->path), ctx->max_num_threads(), in TrMul()
|
| A D | ctx.cc | 44 int Ctx::max_num_threads() const { return impl().max_num_threads_; } in max_num_threads() function in ruy::Ctx
|
| A D | trace.h | 606 tentative_thread_count, rows, depth, cols, ctx->max_num_threads());
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/ |
| 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());
|
| 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
|
| A D | cpu_backend_gemm_custom_gemv.h | 165 context->max_num_threads(), dst_params.rows, dst_params.cols, in CustomGemv()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/ |
| A D | external_cpu_backend_context.h | 36 virtual void SetMaxNumThreads(int max_num_threads) = 0;
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
| A D | depthwiseconv_multithread.h | 133 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConv()
|
| A D | legacy_optimized_ops.h | 464 gemmlowp_context ? gemmlowp_context->max_num_threads() : 1;
566 gemmlowp_context ? gemmlowp_context->max_num_threads() : 1;
1057 gemmlowp_context->max_num_threads(), output_rows, batches, input_size); in FullyConnectedAsGEMV()
2068 gemmlowp_context->max_num_threads(), output_rows, batches, input_size); in LegacyInt8FullyConnectedAsGEMV()
2307 gemmlowp_context->max_num_threads(), output_depth, batches, accum_depth); in ShuffledFullyConnected()
|
| A D | optimized_ops.h | 903 LegacyHowManyThreads<kKernelRows>(cpu_backend_context->max_num_threads(), in ShuffledFullyConnected()
1167 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
3717 : std::min(thread_count, cpu_backend_context->max_num_threads());
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/ |
| A D | common.h | 976 inline int LegacyHowManyThreads(int max_num_threads, int rows, int cols, in LegacyHowManyThreads() argument
979 if (max_num_threads == 1) { in LegacyHowManyThreads()
984 int thread_count = std::min(max_num_threads, rows / KernelRows); in LegacyHowManyThreads()
1003 assert(thread_count > 0 && thread_count <= max_num_threads); in LegacyHowManyThreads()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| A D | fully_connected.h | 174 const int max_threads = cpu_backend_context->max_num_threads(); in FullyConnectedSparseWeight1x4()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| A D | mean.h | 219 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
|
| A D | depthwise_conv_hybrid.h | 420 const int max_threads = cpu_backend_context->max_num_threads(); in DepthwiseConvHybridPerChannel()
|
| A D | depthwise_conv.h | 1986 const int max_threads = cpu_backend_context->max_num_threads();
|