Set thread count for tflite_run (#5830)

Set thread count for tflite_run by using THREAD environment variable

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>

diff --git a/tests/tools/tflite_run/src/tflite_run.cc b/tests/tools/tflite_run/src/tflite_run.cc
index b54f0e1..c067f8b 100644 (file)
--- a/tests/tools/tflite_run/src/tflite_run.cc
+++ b/tests/tools/tflite_run/src/tflite_run.cc
@@ -73,7 +73,7 @@ int main(const int argc, char **argv)
 
       TFLITE_ENSURE(builder(&interpreter))
 
-      interpreter->SetNumThreads(1);
+      interpreter->SetNumThreads(nnfw::misc::get_env_int("THREAD", -1));
     };
   }
   catch (const std::exception &e)