[moco-tf] Summary for TFConv2DBackpropInput (#7575)

This commit introduces summary for TFConv2DBackpropInput.

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 08fad7a..849f8da 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -99,6 +99,7 @@ private:
   IMPLEMENT(TFConcatV2);
   IMPLEMENT(TFConst);
   IMPLEMENT(TFConv2D);
+  IMPLEMENT(TFConv2DBackpropInput);
   IMPLEMENT(TFDepthwiseConv2dNative);
   IMPLEMENT(TFFusedBatchNorm);
   IMPLEMENT(TFMaxPool);
@@ -209,6 +210,18 @@ bool TFNodeSummaryBuilder::summary(const TFConv2D *node, locop::NodeSummary &s)
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFConv2DBackpropInput *node, locop::NodeSummary &s) const
+{
+  s.args().append("input_sizes", tbl()->lookup(node->input_sizes()));
+  s.args().append("filter", tbl()->lookup(node->filter()));
+  s.args().append("out_backprop", tbl()->lookup(node->out_backprop()));
+  s.args().append("padding", node->padding());
+  s.args().append("data_layout", node->data_layout());
+  s.args().append("strides", pepper::str(node->strides()));
+  s.state(locop::NodeSummary::State::PartiallyKnown);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFDepthwiseConv2dNative *node, locop::NodeSummary &s) const
 {
   s.args().append("input", tbl()->lookup(node->input()));