Home
last modified time | relevance | path

Searched refs:SetNumThreads (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Deigen_support.cc74 SetNumThreads(num_threads); in LazyEigenThreadPoolHolder()
89 void SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::eigen_support::__anon64c280760111::LazyEigenThreadPoolHolder
122 ptr->thread_pool_holder->SetNumThreads(context->recommended_num_threads); in Refresh()
Dtest_util.h329 void SetNumThreads(int num_threads) { in SetNumThreads() function
330 interpreter_->SetNumThreads(num_threads); in SetNumThreads()
Dconv_test.cc372 m.SetNumThreads(i); in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc264 interpreter_->SetNumThreads(1); in TEST_F()
289 interpreter_->SetNumThreads(4); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dmeta_support.h38 void SetNumThreads(int num_threads);
Dmeta_support.cc196 void SetNumThreads(int num_threads) { g_num_threads = num_threads; } in SetNumThreads() function
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api.cc120 interpreter->SetNumThreads(optional_options->num_threads); in TFL_NewInterpreter()
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc206 void Interpreter::SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::Interpreter
Dinterpreter.h346 void SetNumThreads(int num_threads);
Dinterpreter_test.cc897 interpreter_.SetNumThreads(4); in TEST_F()
901 interpreter_.SetNumThreads(4); in TEST_F()
902 interpreter_.SetNumThreads(5); in TEST_F()
907 interpreter_.SetNumThreads(4); in TEST_F()
Dmodel.cc519 (*interpreter)->SetNumThreads(num_threads); in operator ()()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image.cc136 interpreter->SetNumThreads(s->number_of_threads); in RunInference()
/external/tensorflow/tensorflow/lite/examples/ios/simple/
DRunModelViewController.mm124 interpreter->SetNumThreads(num_threads);
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dfaq.md118 multi-threaded kernels. You can use `SetNumThreads()` in the
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc263 interpreter->SetNumThreads(static_cast<int>(num_threads)); in Java_org_tensorflow_lite_NativeInterpreterWrapper_numThreads()