From: 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Sun, 6 Oct 2019 23:46:17 +0000 (+0900)
Subject: [exo-tflite] Complete node summary for TFLAvergePool2D & TFLMaxPool2D (#7928)
X-Git-Tag: submit/tizen/20191205.083104~944
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a75cd15fc67f2bd4ba47900ddf3f297f7401b0ea;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] Complete node summary for TFLAvergePool2D & TFLMaxPool2D (#7928)

Complete node summary for TFLAvergePool2D & TFLMaxPool2D was added.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---

diff --git a/compiler/exo-tflite/src/TFLFormattedGraph.cpp b/compiler/exo-tflite/src/TFLFormattedGraph.cpp
index 87acdd9..63464da 100644
--- a/compiler/exo-tflite/src/TFLFormattedGraph.cpp
+++ b/compiler/exo-tflite/src/TFLFormattedGraph.cpp
@@ -61,6 +61,11 @@ std::string to_str(const locoex::Stride *stride)
   return pepper::str(stride->h(), ",", stride->w());
 }
 
+std::string to_str(const locoex::Filter *filter)
+{
+  return pepper::str(filter->h(), ",", filter->w());
+}
+
 std::string opname(uint32_t opnum)
 {
   static std::string prefix{"tfl."};
@@ -155,8 +160,17 @@ bool TFLNodeSummaryBuilder::summary(const locoex::TFLAdd *node, locop::NodeSumma
 bool TFLNodeSummaryBuilder::summary(const locoex::TFLAveragePool2D *node,
                                     locop::NodeSummary &s) const
 {
-  s.args().append("input", tbl()->lookup(node->value()));
-  s.state(locop::NodeSummary::State::PartiallyKnown);
+  auto f = node->fusedActivationFunction();
+  assert(f != locoex::FusedActFunc::UNDEFINED);
+
+  s.args().append("value", tbl()->lookup(node->value()));
+  s.args().append("filter(h,w)", to_str(node->filter()));
+  s.args().append("stride(h,w)", to_str(node->stride()));
+  s.args().append("padding", to_str(node->padding()));
+  s.args().append("fused", to_str(node->fusedActivationFunction()));
+
+  s.state(locop::NodeSummary::State::Complete);
+
   return true;
 }
 
@@ -199,8 +213,17 @@ bool TFLNodeSummaryBuilder::summary(const locoex::TFLDiv *node, locop::NodeSumma
 
 bool TFLNodeSummaryBuilder::summary(const locoex::TFLMaxPool2D *node, locop::NodeSummary &s) const
 {
-  s.args().append("input", tbl()->lookup(node->value()));
-  s.state(locop::NodeSummary::State::PartiallyKnown);
+  auto f = node->fusedActivationFunction();
+  assert(f != locoex::FusedActFunc::UNDEFINED);
+
+  s.args().append("value", tbl()->lookup(node->value()));
+  s.args().append("filter(h,w)", to_str(node->filter()));
+  s.args().append("stride(h,w)", to_str(node->stride()));
+  s.args().append("padding", to_str(node->padding()));
+  s.args().append("fused", to_str(node->fusedActivationFunction()));
+
+  s.state(locop::NodeSummary::State::Complete);
+
   return true;
 }