From 864b9dcfc60feca2391ec96df49f898b38566f8b Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?=
 =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <saehie.park@samsung.com>
Date: Thu, 1 Aug 2019 16:57:29 +0900
Subject: [PATCH] [loco] TypeForward for EltwiseAdd and EltwiseMul (#6101)

This will add TypeForward for EltwiseAdd and EltwiseMul node

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---
 compiler/loco/src/Service/TypeInference.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp
index a993938..6eb9f6f 100644
--- a/compiler/loco/src/Service/TypeInference.cpp
+++ b/compiler/loco/src/Service/TypeInference.cpp
@@ -100,6 +100,8 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<l
   loco::DataType visit(const loco::Conv2D *node) { return loco::dtype_get(node->ifm()); }
   // TODO Support DepthwiseConv2D
   // TODO Support DepthwiseFilterEncode
+  loco::DataType visit(const loco::EltwiseAdd *node) { return loco::dtype_get(node->lhs()); }
+  loco::DataType visit(const loco::EltwiseMul *node) { return loco::dtype_get(node->lhs()); }
   // TODO Support Forward
   loco::DataType visit(const loco::FeatureBiasAdd *node) { return loco::dtype_get(node->value()); }
   loco::DataType visit(const loco::FeatureDecode *node) { return loco::dtype_get(node->input()); }
-- 
2.7.4