Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Deigen_support.cc118 SetEigenNbThreads(context->recommended_num_threads); in Refresh()
122 ptr->thread_pool_holder->SetNumThreads(context->recommended_num_threads); in Refresh()
133 if (context->recommended_num_threads != -1) { in IncrementUsageCounter()
134 SetEigenNbThreads(context->recommended_num_threads); in IncrementUsageCounter()
140 new LazyEigenThreadPoolHolder(context->recommended_num_threads)); in IncrementUsageCounter()
Dgemm_support.cc38 ptr->gemm_context->set_max_num_threads(context->recommended_num_threads); in Refresh()
52 if (context->recommended_num_threads != -1) { in IncrementUsageCounter()
53 ptr->gemm_context->set_max_num_threads(context->recommended_num_threads); in IncrementUsageCounter()
Dconv.cc261 data->run_multithreaded_kernel = context->recommended_num_threads != 1; in Prepare()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate.cc36 if (context->recommended_num_threads > 0) { in Prepare()
38 context->recommended_num_threads); in Prepare()
/external/tensorflow/tensorflow/lite/experimental/micro/testing/
Dtest_utils.h92 context->recommended_num_threads = 1; in PopulateContext()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.h448 int recommended_num_threads; member
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc208 subgraph->context()->recommended_num_threads = num_threads; in SetNumThreads()
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.cc153 context_.recommended_num_threads = 1; in MicroInterpreter()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc154 context_->recommended_num_threads = -1; in Subgraph()