[moco-tf] TFNodeSummaryBuilder for TFConcatV2 (#6727)

This will add TFNodeSummaryBuilder summary method for TFConcatV2

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index fb64544..d0d6a16 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -78,6 +78,7 @@ private:
   IMPLEMENT(TFAdd);
   IMPLEMENT(TFAvgPool);
   IMPLEMENT(TFBiasAdd);
+  IMPLEMENT(TFConcatV2);
   IMPLEMENT(TFConst);
   IMPLEMENT(TFConv2D);
   IMPLEMENT(TFDepthwiseConv2dNative);
@@ -140,6 +141,20 @@ bool TFNodeSummaryBuilder::summary(const TFBiasAdd *node, locop::NodeSummary &s)
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFConcatV2 *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.ConcatV2");
+  for (uint32_t n = 0; n < node->num_values(); ++n)
+  {
+    std::ostringstream ss;
+    ss << "values(" << n << ")";
+    s.args().append(ss.str(), tbl()->lookup(node->values(n)));
+  }
+  s.args().append("axis", tbl()->lookup(node->axis()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFConst *node, locop::NodeSummary &s) const
 {
   std::ostringstream ss;