[tflite_benchmark] Show benchmark configurations (#1491)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Fri, 1 Jun 2018 01:05:30 +0000 (10:05 +0900)
committer서상민/동작제어Lab(SR)/Staff Engineer/삼성전자 <sangmin7.seo@samsung.com>
Fri, 1 Jun 2018 01:05:30 +0000 (10:05 +0900)
This commit revises tflite_benchmark to show benchmark configurations
before running benchmark.

Currently, the following configurations will be shown:
 - The number of threads to be used
 - Whether NNAPI is used, or not

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
tools/tflite_benchmark/src/tflite_benchmark.cc

index 4f1f32d..fd1f580 100644 (file)
@@ -87,6 +87,12 @@ int main(const int argc, char **argv)
     use_nnapi = true;
   }
 
+  std::cout << "Num threads: " << thread.value() << std::endl;
+  if (use_nnapi)
+  {
+    std::cout << "Use NNAPI" << std::endl;
+  }
+
   StderrReporter error_reporter;
 
   auto model = FlatBufferModel::BuildFromFile(filename, &error_reporter);