[moco/tf] Add TFNodeSummaryBuilder for DepthwiseConv2dNative (#6004)

This commit adds a TFNodeSummaryBuilder for DepthwiseConv2dNative.

Signed-off-by: seongwoo <sw4670.chae@samsung.com>

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index 10b5fa0..48f5906 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -71,6 +71,7 @@ private:
   IMPLEMENT(TFBiasAdd);
   IMPLEMENT(TFConst);
   IMPLEMENT(TFConv2D);
+  IMPLEMENT(TFDepthwiseConv2dNative);
   IMPLEMENT(TFFusedBatchNorm);
   IMPLEMENT(TFMul);
   IMPLEMENT(TFReshape);
@@ -155,6 +156,22 @@ bool TFNodeSummaryBuilder::summary(const TFConv2D *node, locop::NodeSummary &s)
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFDepthwiseConv2dNative *node, locop::NodeSummary &s) const
+{
+  std::ostringstream ss;
+
+  s.opname("TF.DepthwiseConv2dNative");
+  s.args().append("ifm", _tbl->lookup(node->ifm()));
+  s.args().append("ker", _tbl->lookup(node->ker()));
+  s.args().append("padding", node->padding());
+  s.args().append("data_layout", node->data_layout());
+
+  ss << node->strides();
+  s.args().append("strides", ss.str());
+  s.state(locop::NodeSummary::State::PartiallyKnown);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFFusedBatchNorm *node, locop::NodeSummary &s) const
 {
   std::ostringstream ss;