[moco-tf] Logging compile stage (#6116)

This commit introduce logging onset of each compile stage.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

diff --git a/compiler/moco-tf/src/Frontend.cpp b/compiler/moco-tf/src/Frontend.cpp
index 487360e..75b1a69 100644 (file)
--- a/compiler/moco-tf/src/Frontend.cpp
+++ b/compiler/moco-tf/src/Frontend.cpp
@@ -15,6 +15,7 @@
  */
 
 #include <moco/tf/Frontend.h>
+#include <moco/Log.h>
 
 #include "Canonicalizer.h"
 #include "Importer.h"
@@ -222,16 +223,24 @@ void cleanup(loco::Graph *graph)
 std::unique_ptr<loco::Graph> Frontend::import(const ModelSignature &signature,
                                               tensorflow::GraphDef &tf_graph_def) const
 {
+  LOGGER(frontend);
+
   // Let's use GraphBuilderRegistry with COpCallGraphBuilder
   GraphBuilderRegistry registry = make_graph_builder_registry(signature);
 
   Importer importer{&registry};
 
+  INFO(frontend) << ">>";
+  INFO(frontend) << ">> Import stage started";
+  INFO(frontend) << ">>";
   auto graph = importer.import(signature, tf_graph_def);
 
   TFOptimizer tfoptimizier;
 
   // Transform TFNodes
+  INFO(frontend) << ">>";
+  INFO(frontend) << ">> TF optimize stage started";
+  INFO(frontend) << ">>";
   tfoptimizier.optimize(graph.get());
 
   // Fill graph-level input/output shape
@@ -252,11 +261,17 @@ std::unique_ptr<loco::Graph> Frontend::import(const ModelSignature &signature,
   // Convert graph to hold only Canonical dialect
   Canonicalizer canonicalizer;
 
+  INFO(frontend) << ">>";
+  INFO(frontend) << ">> Canonicalize stage started";
+  INFO(frontend) << ">>";
   canonicalizer.canonicalize(graph.get());
 
   // Optimize imported loco::Graph
   Optimizer optimizer;
 
+  INFO(frontend) << ">>";
+  INFO(frontend) << ">> Canonical optimize stage started";
+  INFO(frontend) << ">>";
   optimizer.optimize(graph.get());
 
   return std::move(graph);