[moco-tf] Summary for TFSqueeze (#6162)

* [moco-tf] Summary for TFSqueeze

This commit introduces log summary for TFSqueeze

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
* Make use of pepper::str

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index 959be64..251c7d4 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -81,6 +81,7 @@ private:
   IMPLEMENT(TFMul);
   IMPLEMENT(TFReshape);
   IMPLEMENT(TFRsqrt);
+  IMPLEMENT(TFSqueeze);
 #undef IMPLEMENT
 };
 
@@ -234,6 +235,15 @@ bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) c
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFSqueeze *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Squeeze");
+  s.args().append("input", _tbl->lookup(node->input()));
+  s.args().append("squeeze_dims", pepper::str(node->squeeze_dims()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 } // namespace
 
 namespace moco