[exo-tflite] Add EltwiseConverter to conversion list (#7463)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 16 Sep 2019 08:15:37 +0000 (17:15 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Mon, 16 Sep 2019 08:15:37 +0000 (17:15 +0900)
This will add EltwiseConverter to conversion list from knob status

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
compiler/exo-tflite/src/Conversions.h
compiler/exo-tflite/src/Convert.cpp

index 54c8584..26a6258 100644 (file)
@@ -22,7 +22,7 @@
 // TODO loco::Conv2D
 // TODO loco::DepthwiseConv2D
 // TODO loco::DepthwiseFilterEncode
-// TODO loco::EltwiseAdd
+#include "Conversion/EltwiseAddConverter.h"
 // TODO loco::EltwiseDiv
 // TODO loco::EltwiseMul
 // TODO loco::EltwiseSqrt
index 6f5ffe8..267cffa 100644 (file)
@@ -62,7 +62,9 @@ void convert_to_TFLNodes(loco::Graph *graph)
     // TODO loco::Conv2D
     // TODO loco::DepthwiseConv2D
     // TODO loco::DepthwiseFilterEncode
-    // TODO loco::EltwiseAdd
+    if (get_knob<Knob::ConvertEltwiseAdd>())
+      phase.emplace_back(stdex::make_unique<EltwiseAddConverter>());
+
     // TODO loco::EltwiseDiv
     // TODO loco::EltwiseMul
     // TODO loco::EltwiseSqrt