From fb28a5f8e2e78b694f2ada8527f16f7508744dff Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=82=A8=EA=B6=81=EC=84=9D/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 19 Aug 2019 08:30:57 +0900 Subject: [PATCH] [locop] Add TensorSoftmax visit function (#6648) This commit will add `TensorSoftmax` visit function Signed-off-by: Seok NamKoong --- compiler/locop/src/CanonicalNodeSummaryBuilder.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp index 2691f10..651e6fe 100644 --- a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp +++ b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp @@ -290,6 +290,17 @@ public: return res; } + NodeDesc visit(const loco::TensorSoftmax *node) final + { + NodeDesc res{opname(node)}; + + res.arg("input", symbol_lookup(*_symtbl, node->input())); + res.arg("axis", pp::fmt(node->axis())); + res.state(NodeDesc::State::Complete); + + return res; + } + public: NodeDesc visit(const loco::Node *node) final { return default_node_desc(*_symtbl, node); } -- 2.7.4