From 9267be8715556b2475c9034535ba0d4e30728a88 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 09:59:03 +0900 Subject: [PATCH] [moco-tf] Summary for TFSqueeze (#6162) * [moco-tf] Summary for TFSqueeze This commit introduces log summary for TFSqueeze Signed-off-by: Cheongyo Bahk * Make use of pepper::str --- compiler/moco-tf/src/TFFormattedGraph.cpp | 10 ++++++++++ 1 file changed, 10 insertions(+) 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 -- 2.7.4