From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Mon, 23 Sep 2019 01:32:23 +0000 (+0900)
Subject: [exo-tflite] Enable Div,Sub conversion (#7669)
X-Git-Tag: submit/tizen/20191205.083104~1123
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=f525e33b07759e983110786708cae1498156c599;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] Enable Div,Sub conversion (#7669)

This will enable Eltwise Div/Sub node conversion to TFL node by knob

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

diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h
index 6e88063..e9e3447 100644
--- a/compiler/exo-tflite/src/Conversions.h
+++ b/compiler/exo-tflite/src/Conversions.h
@@ -23,10 +23,10 @@
 // TODO loco::DepthwiseConv2D
 // TODO loco::DepthwiseFilterEncode
 #include "Conversion/EltwiseAddConverter.h"
-// TODO loco::EltwiseDiv
+#include "Conversion/EltwiseDivConverter.h"
 #include "Conversion/EltwiseMulConverter.h"
 // TODO loco::EltwiseSqrt
-// TODO loco::EltwiseSub
+#include "Conversion/EltwiseSubConverter.h"
 #include "Conversion/FeatureBiasAddConverter.h"
 // TODO loco::FixedReshape
 #include "Conversion/MaxPool2DConverter.h"
diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp
index 516227d..9883c6a 100644
--- a/compiler/exo-tflite/src/Convert.cpp
+++ b/compiler/exo-tflite/src/Convert.cpp
@@ -67,12 +67,16 @@ void convert_to_TFLNodes(loco::Graph *graph)
     if (get<Knob::ConvertEltwiseAdd>())
       phase.emplace_back(stdex::make_unique<EltwiseAddConverter>());
 
-    // TODO loco::EltwiseDiv
+    if (get<Knob::ConvertEltwiseDiv>())
+      phase.emplace_back(stdex::make_unique<EltwiseDivConverter>());
+
     if (get<Knob::ConvertEltwiseMul>())
       phase.emplace_back(stdex::make_unique<EltwiseMulConverter>());
 
     // TODO loco::EltwiseSqrt
-    // TODO loco::EltwiseSub
+
+    if (get<Knob::ConvertEltwiseSub>())
+      phase.emplace_back(stdex::make_unique<EltwiseSubConverter>());
 
     if (get<Knob::ConvertFeatureBiasAdd>())
       phase.emplace_back(stdex::make_unique<FeatureBiasAddConverter>());