[moco-tf] Support TFAvgPool for Logging (#6045)

This will add TFAvgPool for TFNodeSummaryBuilder thus dump of graph shows TFAvgPool attributes

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 48f5906..89975a6 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -68,6 +68,7 @@ public:
 private:
 #define IMPLEMENT(CLASS) bool summary(const CLASS *, locop::NodeSummary &) const final
   IMPLEMENT(TFAdd);
+  IMPLEMENT(TFAvgPool);
   IMPLEMENT(TFBiasAdd);
   IMPLEMENT(TFConst);
   IMPLEMENT(TFConv2D);
@@ -104,6 +105,25 @@ bool TFNodeSummaryBuilder::summary(const TFAdd *node, locop::NodeSummary &s) con
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFAvgPool *node, locop::NodeSummary &s) const
+{
+  std::ostringstream ss;
+
+  s.opname("TF.AvgPool");
+  s.args().append("value", _tbl->lookup(node->value()));
+
+  ss << node->ksize();
+  s.args().append("ksize", ss.str());
+  ss.str("");
+  ss << node->strides();
+  s.args().append("strides", ss.str());
+  s.args().append("padding", node->padding());
+  s.args().append("data_layout", node->data_layout());
+
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFBiasAdd *node, locop::NodeSummary &s) const
 {
   s.opname("TF.BiasAdd");