[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>

diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h
index 26a6258..8f2e428 100644 (file)
--- 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 (file)
--- 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