From 2053aeee933e71dede2718c36bd21d2e8fbaafcf Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/On-Device=20Lab=28SR=29/Staff?= =?utf8?q?=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 18:38:03 +0900 Subject: [PATCH] [loco] Implement ReLU6 shape inference (#6222) This commit extends CanonicalShapeInferenceRule to accept ReLU6 node. Signed-off-by: Jonghyun Park --- compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 }; -- 2.7.4