From: 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Mon, 16 Sep 2019 00:35:16 +0000 (+0900)
Subject: [exo-tflite] Adding MaxPool2DConverter into convert.cpp (#7400)
X-Git-Tag: accepted/tizen/unified/20190918.102349~96
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=43c0a7f5c96f439cc9c207653b4068e388526014;p=platform%2Fcore%2Fml%2Fnnfw.git

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