From: 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Date: Mon, 5 Aug 2019 09:38:03 +0000 (+0900)
Subject: [loco] Implement ReLU6 shape inference (#6222)
X-Git-Tag: submit/tizen/20190809.050447~165
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=2053aeee933e71dede2718c36bd21d2e8fbaafcf;p=platform%2Fcore%2Fml%2Fnnfw.git

[loco] Implement ReLU6 shape inference (#6222)

This commit extends CanonicalShapeInferenceRule to accept ReLU6 node.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---

diff --git a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
index 348e54f..ae1df7b 100644
--- a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
+++ b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
@@ -95,7 +95,9 @@ public:
   // CASE: ReLU
   loco::NodeShape visit(const loco::ReLU *node) final { return loco::shape_get(node->input()); }
 
-  // TODO Support ReLU6
+  // CASE: ReLU6
+  loco::NodeShape visit(const loco::ReLU6 *node) final { return loco::shape_get(node->input()); }
+
   // TODO Support TensorBiasAdd
   // TODO SUpport TensorConcat
 };