add custom number of threads support

Change-Id: I3d550c5921dba536dc113324c0ec25ce37150539
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp
index 948a1c441e3d948c7369dab60d08c1d5cc5b8b5e..81cdbdbb2ff04d82f01a0d8dc9ee5176dc148d29 100644 (file)
--- a/src/inference_engine_tflite.cpp
+++ b/src/inference_engine_tflite.cpp
@@ -131,7 +131,12 @@ namespace TFLiteImpl
                        }
                }
 
-               mInterpreter->SetNumThreads(MV_INFERENCE_TFLITE_MAX_THREAD_NUM);
+               const char *envNumThreads = std::getenv("TFLITE_NUM_THREADS");
+               int numThreads = envNumThreads ? std::atoi(envNumThreads) : MV_INFERENCE_TFLITE_MAX_THREAD_NUM;
+
+               LOGD("numThreads : [%d]", numThreads);
+
+               mInterpreter->SetNumThreads(numThreads);
                LOGI("mInterpreter->tensors_size() :[%zu]",
                         mInterpreter->tensors_size());