From: 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Wed, 11 Sep 2019 05:29:09 +0000 (+0900)
Subject: [exo-tflite] TypeInference for TFLMaxPool2D (#7350)
X-Git-Tag: accepted/tizen/unified/20190918.102349~140
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=fc471424f00d971cdd300897f0ea00f4acaf4bad;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] TypeInference for TFLMaxPool2D (#7350)

This adds type inference for `TFLMaxPool2D`.

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

diff --git a/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp b/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp
index 1d349a2..ee9918e 100644
--- a/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp
+++ b/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp
@@ -55,7 +55,10 @@ struct TypeInferenceAlgorithm final : public locoex::TFLNodeVisitor<loco::DataTy
 
   // TODO TFLDiv
 
-  // TODO TFLMaxPool2D
+  loco::DataType visit(const locoex::TFLMaxPool2D *node) final
+  {
+    return loco::dtype_get(node->value());
+  }
 
   // TODO TFLMul