[locop] Support EltwiseAdd and EltwiseMul (#6098)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 1 Aug 2019 08:33:17 +0000 (17:33 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 1 Aug 2019 08:33:17 +0000 (17:33 +0900)
* [locop] Support EltwiseAdd and EltwiseMul

This will add locop to support EltwiseAdd and EltwiseMul to dump it's contents

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
* remove todo msg

compiler/locop/src/FormattedGraph.cpp

index b6808ab..b2d78f4 100644 (file)
@@ -289,6 +289,28 @@ public:
     return res;
   }
 
+  NodeDesc visit(const loco::EltwiseAdd *node) final
+  {
+    NodeDesc res{opname(node)};
+
+    res.arg("lhs", symbol_lookup(*_symtbl, node->lhs()));
+    res.arg("rhs", symbol_lookup(*_symtbl, node->rhs()));
+    res.state(NodeDesc::State::Complete);
+
+    return res;
+  }
+
+  NodeDesc visit(const loco::EltwiseMul *node) final
+  {
+    NodeDesc res{opname(node)};
+
+    res.arg("lhs", symbol_lookup(*_symtbl, node->lhs()));
+    res.arg("rhs", symbol_lookup(*_symtbl, node->rhs()));
+    res.state(NodeDesc::State::Complete);
+
+    return res;
+  }
+
 public:
   NodeDesc visit(const loco::Node *node) final { return default_node_desc(*_symtbl, node); }