diff options
author | 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com> | 2019-09-16 15:57:42 +0900 |
---|---|---|
committer | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-16 15:57:42 +0900 |
commit | b363925c38c87de3cbf665db5b954c61ea7ef293 (patch) | |
tree | f9321e350a829eabcf67d599cccef20ef80c597b /compiler | |
parent | 6b9f7cb9be3b70bbd0509031c484dd492e20274e (diff) | |
download | nnfw-b363925c38c87de3cbf665db5b954c61ea7ef293.tar.gz nnfw-b363925c38c87de3cbf665db5b954c61ea7ef293.tar.bz2 nnfw-b363925c38c87de3cbf665db5b954c61ea7ef293.zip |
[loco] Adding type inference for loco::BiasDecode (#7455)
This adds type inference for loco::BiasDecode.
Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
Diffstat (limited to 'compiler')
-rw-r--r-- | compiler/loco/src/Service/TypeInference.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp index a25d45d6e..3c3c5ad2c 100644 --- a/compiler/loco/src/Service/TypeInference.cpp +++ b/compiler/loco/src/Service/TypeInference.cpp @@ -117,6 +117,7 @@ namespace struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<loco::DataType> { loco::DataType visit(const loco::AvgPool2D *node) { return loco::dtype_get(node->ifm()); } + loco::DataType visit(const loco::BiasDecode *node) { return loco::dtype_get(node->input()); } loco::DataType visit(const loco::BiasEncode *node) { return loco::dtype_get(node->input()); } loco::DataType visit(const loco::ConstGen *node) { return node->dtype(); } loco::DataType visit(const loco::Conv2D *node) { return loco::dtype_get(node->ifm()); } |