From 3cd9b55a41bc6cdc6434fce513ca815f3ed88947 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: Fri, 23 Aug 2019 10:00:57 +0900 Subject: [PATCH] [locop] Add Tanh visit function (#6851) This commit will add `Tanh` visit function Signed-off-by: Seok NamKoong --- compiler/locop/src/CanonicalNodeSummaryBuilder.cpp | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp index 651e6fe..098a9d9 100644 --- a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp +++ b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp @@ -290,6 +290,16 @@ public: return res; } + NodeDesc visit(const loco::Tanh *node) final + { + NodeDesc res{opname(node)}; + + res.arg("input", symbol_lookup(*_symtbl, node->input())); + res.state(NodeDesc::State::Complete); + + return res; + } + NodeDesc visit(const loco::TensorSoftmax *node) final { NodeDesc res{opname(node)}; -- 2.7.4