[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 (file)
--- 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
 };