[moco-tf] TFShape summary (#6372)

This commit adds TFShape node summary.

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 d7a0590..9595270 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -86,6 +86,7 @@ private:
   IMPLEMENT(TFMul);
   IMPLEMENT(TFReshape);
   IMPLEMENT(TFRsqrt);
+  IMPLEMENT(TFShape);
   IMPLEMENT(TFSqueeze);
 #undef IMPLEMENT
 };
@@ -240,6 +241,14 @@ bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) c
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFShape *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Shape");
+  s.args().append("input", tbl()->lookup(node->input()));
+  s.state(locop::NodeSummary::State::PartiallyKnown);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFSqueeze *node, locop::NodeSummary &s) const
 {
   s.opname("TF.Squeeze");