diff options
author | Hyeongseok Oh <hseok82.oh@samsung.com> | 2023-09-08 10:51:25 +0000 |
---|---|---|
committer | Hyeongseok Oh <hseok82.oh@samsung.com> | 2023-09-08 10:51:25 +0000 |
commit | eed258505ee1ad0f72d9e0a8a3934f2e9e7b5e79 (patch) | |
tree | 1aa860656489469003375a0f67edb1d729f7dc6b /compiler/luci/service/src/CircleTypeInferenceRule.cpp | |
parent | 3a0ad354832744d138b361ffcfd21f33494beb6b (diff) | |
download | nnfw-eed258505ee1ad0f72d9e0a8a3934f2e9e7b5e79.tar.gz nnfw-eed258505ee1ad0f72d9e0a8a3934f2e9e7b5e79.tar.bz2 nnfw-eed258505ee1ad0f72d9e0a8a3934f2e9e7b5e79.zip |
Imported Upstream version 1.25.0upstream/1.25.0submit/tizen/20230908.105404upstream
Diffstat (limited to 'compiler/luci/service/src/CircleTypeInferenceRule.cpp')
-rw-r--r-- | compiler/luci/service/src/CircleTypeInferenceRule.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/compiler/luci/service/src/CircleTypeInferenceRule.cpp b/compiler/luci/service/src/CircleTypeInferenceRule.cpp index 44c93305b..bd3feb977 100644 --- a/compiler/luci/service/src/CircleTypeInferenceRule.cpp +++ b/compiler/luci/service/src/CircleTypeInferenceRule.cpp @@ -172,10 +172,20 @@ struct TypeInferenceAlgorithm final : public luci::CircleNodeVisitor<loco::DataT return luci::dtype_get(node->params()); } + loco::DataType visit(const luci::CircleGelu *node) final + { + return luci::dtype_get(node->features()); + } + loco::DataType visit(const luci::CircleGreater *) final { return loco::DataType::BOOL; } loco::DataType visit(const luci::CircleGreaterEqual *) final { return loco::DataType::BOOL; } + loco::DataType visit(const luci::CircleHardSwish *node) final + { + return luci::dtype_get(node->features()); + } + loco::DataType visit(const luci::CircleIf *node) final { // Type of If is not used. Just use input 0 |