From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Tue, 13 Aug 2019 03:51:51 +0000 (+0900)
Subject: [loco] Inference service for EltwiseSqrt (#6526)
X-Git-Tag: accepted/tizen/unified/20190903.052428~444
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=64fd5655c8c0e650b81aa42ecf9a5720e0e1ee27;p=platform%2Fcore%2Fml%2Fnnfw.git

[loco] Inference service for EltwiseSqrt (#6526)

This will add Shape and Type inference for EltwiseSqrt

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---

diff --git a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
index febddb6..448c237 100644
--- a/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
+++ b/compiler/loco/src/Service/CanonicalShapeInferenceRule.cpp
@@ -257,6 +257,12 @@ public:
     return eltwise_binary_node_shape(node);
   }
 
+  // CASE: EltwiseSqrt
+  loco::NodeShape visit(const loco::EltwiseSqrt *node) final
+  {
+    return loco::shape_get(node->input());
+  }
+
   // CASE: EltwiseSub
   loco::NodeShape visit(const loco::EltwiseSub *node) final
   {
diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp
index 40d77b2..b238b4a 100644
--- a/compiler/loco/src/Service/TypeInference.cpp
+++ b/compiler/loco/src/Service/TypeInference.cpp
@@ -106,6 +106,7 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<l
   loco::DataType visit(const loco::EltwiseAdd *node) { return loco::dtype_get(node->lhs()); }
   loco::DataType visit(const loco::EltwiseDiv *node) { return loco::dtype_get(node->lhs()); }
   loco::DataType visit(const loco::EltwiseMul *node) { return loco::dtype_get(node->lhs()); }
+  loco::DataType visit(const loco::EltwiseSqrt *node) { return loco::dtype_get(node->input()); }
   loco::DataType visit(const loco::EltwiseSub *node) { return loco::dtype_get(node->lhs()); }
   // TODO Support Forward
   loco::DataType visit(const loco::FeatureBiasAdd *node) { return loco::dtype_get(node->value()); }