[locop] Add Tanh visit function (#6851)
author남궁석/On-Device Lab(SR)/Engineer/삼성전자 <sk.namkoong@samsung.com>
Fri, 23 Aug 2019 01:00:57 +0000 (10:00 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Fri, 23 Aug 2019 01:00:57 +0000 (10:00 +0900)
This commit will add `Tanh` visit function

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
compiler/locop/src/CanonicalNodeSummaryBuilder.cpp

index 651e6fe..098a9d9 100644 (file)
@@ -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)};