From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Tue, 6 Aug 2019 03:53:31 +0000 (+0900)
Subject: [exo-tflite] Shape inf. for EltwiseSub and EltwiseDiv (#6264)
X-Git-Tag: submit/tizen/20190809.050447~126
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=75bff3133702e1cd19b4ddf52c6c06f9407064b4;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] Shape inf. for EltwiseSub and EltwiseDiv (#6264)

This will enable shape inference of EltwiseSub and EltwiseDiv node

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

diff --git a/compiler/exo-tflite/src/ShapeInference.cpp b/compiler/exo-tflite/src/ShapeInference.cpp
index 4d37440..f14486d 100644
--- a/compiler/exo-tflite/src/ShapeInference.cpp
+++ b/compiler/exo-tflite/src/ShapeInference.cpp
@@ -103,6 +103,8 @@ public:
   NODE(FeatureBiasAdd)
   NODE(EltwiseAdd)
   NODE(EltwiseMul)
+  NODE(EltwiseSub)
+  NODE(EltwiseDiv)
 #undef NODE
   // TODO Put all the visit method implementations inside this class declaration
   ShapeDescription visit(loco::ReLU6 *node) { return gd._node_to_shape[node->input()]; }
@@ -469,6 +471,25 @@ ShapeDescription ShapeGetter::visit(loco::EltwiseMul *node)
   return lhs_shape;
 }
 
+ShapeDescription ShapeGetter::visit(loco::EltwiseSub *node)
+{
+  const ShapeDescription &lhs_shape = gd._node_to_shape[node->lhs()];
+  const ShapeDescription &rhs_shape = gd._node_to_shape[node->rhs()];
+
+  assert(lhs_shape._dims == rhs_shape._dims);
+
+  return lhs_shape;
+}
+
+ShapeDescription ShapeGetter::visit(loco::EltwiseDiv *node)
+{
+  const ShapeDescription &lhs_shape = gd._node_to_shape[node->lhs()];
+  const ShapeDescription &rhs_shape = gd._node_to_shape[node->rhs()];
+
+  assert(lhs_shape._dims == rhs_shape._dims);
+
+  return lhs_shape;
+}
 } // namespace
 
 namespace