[exo-tflite] shape inference for TFLMaxPool2D (#7374)

Adding shape inference for TFLMaxPool2D

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

diff --git a/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp b/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp
index 0f82185..4b636dd 100644 (file)
--- a/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp
+++ b/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp
@@ -111,7 +111,10 @@ public:
 
   // TODO TFLDiv
 
-  // TODO TFLMaxPool2D
+  loco::NodeShape visit(const locoex::TFLMaxPool2D *node) final
+  {
+    return infer_pool_2d_shape(node);
+  }
 
   // TODO TFLMul