diff options
author | 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com> | 2019-09-11 14:29:09 +0900 |
---|---|---|
committer | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-09-11 14:29:09 +0900 |
commit | fc471424f00d971cdd300897f0ea00f4acaf4bad (patch) | |
tree | f6e97975cde6e815ea0d7ce7480f78f7708645c0 /compiler/exo-tflite | |
parent | a42bb7291f60039d7c0a216a2ec481039c978c40 (diff) | |
download | nnfw-fc471424f00d971cdd300897f0ea00f4acaf4bad.tar.gz nnfw-fc471424f00d971cdd300897f0ea00f4acaf4bad.tar.bz2 nnfw-fc471424f00d971cdd300897f0ea00f4acaf4bad.zip |
[exo-tflite] TypeInference for TFLMaxPool2D (#7350)
This adds type 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/TFLTypeInferenceRule.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp b/compiler/exo-tflite/src/Dialect/Service/TFLTypeInferenceRule.cpp index 1d349a26d..ee9918eb3 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 |