From 019a0cd293b2c698c6ca3c1085450b6e820a43f8 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: Tue, 17 Sep 2019 09:40:18 +0900
Subject: [PATCH] [exo-tflite] Add EltwiseMul and TensorBroadcast to Convert
 list (#7474)

This will update conversion list to convert EltwiseMul and TensorBroadcast node

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---
 compiler/exo-tflite/src/Conversions.h | 3 ++-
 compiler/exo-tflite/src/Convert.cpp   | 7 ++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h
index 26a6258..8f2e428 100644
--- a/compiler/exo-tflite/src/Conversions.h
+++ b/compiler/exo-tflite/src/Conversions.h
@@ -24,7 +24,7 @@
 // TODO loco::DepthwiseFilterEncode
 #include "Conversion/EltwiseAddConverter.h"
 // TODO loco::EltwiseDiv
-// TODO loco::EltwiseMul
+#include "Conversion/EltwiseMulConverter.h"
 // TODO loco::EltwiseSqrt
 // TODO loco::EltwiseSub
 // TODO loco::FeatureBiasAdd
@@ -35,6 +35,7 @@
 // TODO loco::Tanh
 // TODO loco::TensorConcat
 // TODO loco::TensorBiasAdd
+#include "Conversion/TensorBroadcastConverter.h"
 // TODO loco::TensorSoftmax
 
 #endif // __CONVERSIONS_H__
diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp
index 267cffa..20035ff 100644
--- a/compiler/exo-tflite/src/Convert.cpp
+++ b/compiler/exo-tflite/src/Convert.cpp
@@ -66,7 +66,9 @@ void convert_to_TFLNodes(loco::Graph *graph)
       phase.emplace_back(stdex::make_unique<EltwiseAddConverter>());
 
     // TODO loco::EltwiseDiv
-    // TODO loco::EltwiseMul
+    if (get_knob<Knob::ConvertEltwiseMul>())
+      phase.emplace_back(stdex::make_unique<EltwiseMulConverter>());
+
     // TODO loco::EltwiseSqrt
     // TODO loco::EltwiseSub
     // TODO loco::FeatureBiasAdd
@@ -82,6 +84,9 @@ void convert_to_TFLNodes(loco::Graph *graph)
     // TODO loco::Tanh
     // TODO loco::TensorConcat
     // TODO loco::TensorBiasAdd
+    if (get_knob<Knob::ConvertTensorBroadcast>())
+      phase.emplace_back(stdex::make_unique<TensorBroadcastConverter>());
+
     // TODO loco::TensorSoftmax
 
     // Add optimization below
-- 
2.7.4