diff options
| author | 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com> | 2019-09-11 14:28:25 +0900 |
|---|---|---|
| committer | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-09-11 14:28:25 +0900 |
| commit | a42bb7291f60039d7c0a216a2ec481039c978c40 (patch) | |
| tree | c63ad9140298e0260bc7f4e0e8568dad70cc903d /compiler/exo-tflite | |
| parent | ad2f1423b6184a7e948779a72e9cb1cb21430cb7 (diff) | |
| download | nnfw-a42bb7291f60039d7c0a216a2ec481039c978c40.tar.gz nnfw-a42bb7291f60039d7c0a216a2ec481039c978c40.tar.bz2 nnfw-a42bb7291f60039d7c0a216a2ec481039c978c40.zip | |
[exo-tflite] shape inference for TFLMaxPool2D (#7374)
Adding shape inference for TFLMaxPool2D
Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
Diffstat (limited to 'compiler/exo-tflite')
| -rw-r--r-- | compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp b/compiler/exo-tflite/src/Dialect/Service/TFLShapeInferenceRule.cpp index 0f8218515..4b636ddfd 100644 --- 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 |