[locop] FixedReshape node summary (#6354)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Thu, 8 Aug 2019 00:22:43 +0000 (09:22 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 8 Aug 2019 00:22:43 +0000 (09:22 +0900)
This commit adds FixedReshape node summary.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
compiler/locop/src/CanonicalNodeSummaryBuilder.cpp

index c11e4b6..7335a66 100644 (file)
@@ -266,6 +266,17 @@ public:
     return res;
   }
 
+  NodeDesc visit(const loco::Reshape<loco::ReshapeType::Fixed> *node) final
+  {
+    NodeDesc res{opname(node)};
+
+    res.arg("input", symbol_lookup(*_symtbl, node->input()));
+    res.arg("shape", pp::fmt(pretty(tensor_shape(node))));
+    res.state(NodeDesc::State::Complete);
+
+    return res;
+  }
+
 public:
   NodeDesc visit(const loco::Node *node) final { return default_node_desc(*_symtbl, node); }