[tflite_benchmark] Show Input/Output names (#1681)

This commit revises tflite_benchmark to show input/output tensor name to
make it easy to understand the model under analysis.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/tools/tflite_benchmark/src/tflite_benchmark.cc b/tools/tflite_benchmark/src/tflite_benchmark.cc
index fd8d9bc..1ce214d 100644 (file)
--- a/tools/tflite_benchmark/src/tflite_benchmark.cc
+++ b/tools/tflite_benchmark/src/tflite_benchmark.cc
@@ -107,6 +107,28 @@ int main(const int argc, char **argv)
 
   TFLITE_ENSURE(builder(&interpreter));
 
+  // Show inputs
+  for (uint32_t n = 0; n < interpreter->inputs().size(); ++n)
+  {
+    // TODO Print shape
+    auto tensor_id = interpreter->inputs().at(n);
+    auto tensor_ptr = interpreter->tensor(tensor_id);
+
+    std::cout << "Input #" << n << ":" << std::endl;
+    std::cout << "  Name: " << tensor_ptr->name << std::endl;
+  }
+
+  // Show outputs
+  for (uint32_t n = 0; n < interpreter->outputs().size(); ++n)
+  {
+    // TODO Print shape
+    auto tensor_id = interpreter->outputs().at(n);
+    auto tensor_ptr = interpreter->tensor(tensor_id);
+
+    std::cout << "Output #" << n << ":" << std::endl;
+    std::cout << "  Name: " << tensor_ptr->name << std::endl;
+  }
+
   interpreter->SetNumThreads(thread.value());
 
   std::shared_ptr<nnfw::support::tflite::Session> sess;