[moco-tf] Summary for TFReshape node (#5979)

This commit adds summary for TFReshape node

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

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index bb7ff5f..2220498 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -71,6 +71,7 @@ private:
   IMPLEMENT(TFConv2D);
   IMPLEMENT(TFFusedBatchNorm);
   IMPLEMENT(TFMul);
+  IMPLEMENT(TFReshape);
   IMPLEMENT(TFRsqrt);
 #undef IMPLEMENT
 };
@@ -187,6 +188,15 @@ bool TFNodeSummaryBuilder::summary(const TFMul *node, locop::NodeSummary &s) con
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFReshape *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Reshape");
+  s.args().append("tensor", _tbl->lookup(node->tensor()));
+  s.args().append("shape", _tbl->lookup(node->shape()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) const
 {
   s.opname("TF.Rsqrt");