From 3dd9ca9121812fe154b9789969c28229890f1992 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Fri, 1 Jun 2018 10:05:30 +0900 Subject: [PATCH] [tflite_benchmark] Show benchmark configurations (#1491) 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 --- tools/tflite_benchmark/src/tflite_benchmark.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tools/tflite_benchmark/src/tflite_benchmark.cc b/tools/tflite_benchmark/src/tflite_benchmark.cc index 4f1f32d..fd1f580 100644 --- a/tools/tflite_benchmark/src/tflite_benchmark.cc +++ b/tools/tflite_benchmark/src/tflite_benchmark.cc @@ -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); -- 2.7.4