[exo-tflite] Adding MaxPool2DConverter into convert.cpp (#7400)

Now MaxPool2DConverter will work with knob on.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h
index 712a21b..804f9c6 100644 (file)
--- a/compiler/exo-tflite/src/Conversions.h
+++ b/compiler/exo-tflite/src/Conversions.h
@@ -29,7 +29,7 @@
 // TODO loco::EltwiseSub
 // TODO loco::FeatureBiasAdd
 // TODO loco::FixedReshape
-// TODO loco::MaxPool2D
+#include "Conversion/MaxPool2DConverter.h"
 #include "Conversion/ReluConverter.h"
 // TODO loco::ReLU6
 // TODO loco::Tanh

diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp
index c97a64c..11bdd9f 100644 (file)
--- a/compiler/exo-tflite/src/Convert.cpp
+++ b/compiler/exo-tflite/src/Convert.cpp
@@ -62,7 +62,9 @@ void convert_to_TFLNodes(loco::Graph *graph)
     // TODO loco::EltwiseSub
     // TODO loco::FeatureBiasAdd
     // TODO loco::FixedReshape
-    // TODO loco::MaxPool2D
+
+    if (get_knob<Knob::ConvertMaxPool2D>())
+      phase.emplace_back(stdex::make_unique<MaxPool2DConverter>());
 
     if (get_knob<Knob::ConvertRelu>())
       phase.emplace_back(stdex::make_unique<ReluConverter>());