From: 박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Date: Mon, 5 Aug 2019 00:59:03 +0000 (+0900)
Subject: [moco-tf] Summary for TFSqueeze (#6162)
X-Git-Tag: submit/tizen/20190809.050447~200
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=9267be8715556b2475c9034535ba0d4e30728a88;p=platform%2Fcore%2Fml%2Fnnfw.git

[moco-tf] Summary for TFSqueeze (#6162)

* [moco-tf] Summary for TFSqueeze

This commit introduces log summary for TFSqueeze

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

* Make use of pepper::str
---

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index 959be64..251c7d4 100644
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -81,6 +81,7 @@ private:
   IMPLEMENT(TFMul);
   IMPLEMENT(TFReshape);
   IMPLEMENT(TFRsqrt);
+  IMPLEMENT(TFSqueeze);
 #undef IMPLEMENT
 };
 
@@ -234,6 +235,15 @@ bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) c
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFSqueeze *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Squeeze");
+  s.args().append("input", _tbl->lookup(node->input()));
+  s.args().append("squeeze_dims", pepper::str(node->squeeze_dims()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 } // namespace
 
 namespace moco