[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 (file)
--- 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 (file)
--- 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>());