Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Deigen_support_test.cc29 recommended_num_threads = -1; in TestTfLiteContext()
65 context.recommended_num_threads = 1; in TEST()
86 context.recommended_num_threads = 2; in TEST()
106 context.recommended_num_threads = 1; in TEST()
113 context.recommended_num_threads = 3; in TEST()
121 context.recommended_num_threads = -1; in TEST()
128 context.recommended_num_threads = 0; in TEST()
135 context.recommended_num_threads = 3; in TEST()
143 context.recommended_num_threads = -5; in TEST()
Deigen_support.cc140 if (IsValidNumThreads(context->recommended_num_threads)) { in Refresh()
141 SetEigenNbThreads(GetNumThreads(context->recommended_num_threads)); in Refresh()
146 ptr->thread_pool_holder->SetNumThreads(context->recommended_num_threads); in Refresh()
157 if (IsValidNumThreads(context->recommended_num_threads)) { in IncrementUsageCounter()
158 SetEigenNbThreads(context->recommended_num_threads); in IncrementUsageCounter()
164 new LazyEigenThreadPoolHolder(context->recommended_num_threads)); in IncrementUsageCounter()
Dcpu_backend_context.cc117 cpu_backend_context->SetMaxNumThreads(context->recommended_num_threads); in GetFromContext()
Dconv.cc393 (context->recommended_num_threads != 1) && !is_hybrid && in Prepare()
/external/tensorflow/tensorflow/lite/
Dexternal_cpu_backend_context.cc26 context->recommended_num_threads != -1) { in RefreshExternalCpuBackendContext()
28 context->recommended_num_threads); in RefreshExternalCpuBackendContext()
Dinterpreter.cc358 subgraph->context()->recommended_num_threads = num_threads; in SetNumThreads()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate.cc66 if (context->recommended_num_threads > 0) { in Initialize()
68 context->recommended_num_threads); in Initialize()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dkernel_runner.cc44 context_.recommended_num_threads = 1; in KernelRunner()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc54 context.recommended_num_threads = 1; in PopulateContext()
/external/tensorflow/tensorflow/lite/c/
Dcommon.h676 int recommended_num_threads; member
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.cc175 context_.recommended_num_threads = 1; in Init()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc227 context_.recommended_num_threads = -1; in Subgraph()