From cd394e40fe06724a368c0571b6bef1e23cc927e7 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: Thu, 8 Aug 2019 13:00:26 +0900 Subject: [PATCH] [moco-tf] TFShape summary (#6372) This commit adds TFShape node summary. Signed-off-by: Cheongyo Bahk --- compiler/moco-tf/src/TFFormattedGraph.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp index d7a0590..9595270 100644 --- a/compiler/moco-tf/src/TFFormattedGraph.cpp +++ b/compiler/moco-tf/src/TFFormattedGraph.cpp @@ -86,6 +86,7 @@ private: IMPLEMENT(TFMul); IMPLEMENT(TFReshape); IMPLEMENT(TFRsqrt); + IMPLEMENT(TFShape); IMPLEMENT(TFSqueeze); #undef IMPLEMENT }; @@ -240,6 +241,14 @@ bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) c return true; } +bool TFNodeSummaryBuilder::summary(const TFShape *node, locop::NodeSummary &s) const +{ + s.opname("TF.Shape"); + s.args().append("input", tbl()->lookup(node->input())); + s.state(locop::NodeSummary::State::PartiallyKnown); + return true; +} + bool TFNodeSummaryBuilder::summary(const TFSqueeze *node, locop::NodeSummary &s) const { s.opname("TF.Squeeze"); -- 2.7.4