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