Home
last modified time | relevance | path

Searched refs:SetMaxNumThreads (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_context.cc118 cpu_backend_context->SetMaxNumThreads(context->recommended_num_threads); in GetFromContext()
130 SetMaxNumThreads(kDefaultNumThreadpoolThreads); in CpuBackendContext()
141 void CpuBackendContext::SetMaxNumThreads(int max_num_threads) { in SetMaxNumThreads() function in tflite::CpuBackendContext
Dcpu_backend_context.h48 void SetMaxNumThreads(int max_num_threads) override;
Dcpu_backend_threadpool_test.cc69 context.SetMaxNumThreads(num_threads); in TestGenerateArrayOfIncrementingInts()
Dcpu_backend_gemm_test.cc372 cpu_backend_context.SetMaxNumThreads(1 + (random_engine() % 8)); in TestSomeGemm()
528 cpu_backend_context.SetMaxNumThreads(1 + (random_engine() % 8)); in TestMaybeValidGemm()
/external/tensorflow/tensorflow/lite/
Dexternal_cpu_backend_context.cc27 external_context->internal_backend_context()->SetMaxNumThreads( in RefreshExternalCpuBackendContext()
Dexternal_cpu_backend_context.h36 virtual void SetMaxNumThreads(int max_num_threads) = 0;
Dinterpreter_test.cc1230 void SetMaxNumThreads(int num_threads) override {} in SetMaxNumThreads() function
/external/gemmlowp/eight_bit_int_gemm/
Deight_bit_int_gemm.h81 void SetMaxNumThreads(int n);
Deight_bit_int_gemm.cc404 void SetMaxNumThreads(int n) { in SetMaxNumThreads() function
/external/gemmlowp/test/
Dbenchmark_meta_gemm.cc303 gemmlowp::eight_bit_int_gemm::SetMaxNumThreads(4); in main()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc350 backend_context.SetMaxNumThreads(test_param.num_threads); in DispatchDepthwiseConvImpl()
516 backend_context.SetMaxNumThreads(test_param.num_threads); in DispatchDepthwiseConvImpl()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.cc681 cpu_backend_context->SetMaxNumThreads(num_threads); in InitInterpreter()