diff options
| author | 박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com> | 2019-09-17 13:55:15 +0900 |
|---|---|---|
| committer | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-09-17 13:55:15 +0900 |
| commit | 7b5c9ffd644d7d033c82100bc3afc4245ca22443 (patch) | |
| tree | 94c5b90182b9d4170d4cdc493e6294afc7058b90 | |
| parent | fd5551a6187f17f640bea8a0777c313115956839 (diff) | |
| download | nnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.tar.gz nnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.tar.bz2 nnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.zip | |
[loco] Type inference for TransposedConv2D (#7508)
This commit introduces type inference for TransposedConv2D
Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
| -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 3c3c5ad2c..92cdbd51a 100644 --- a/compiler/loco/src/Service/TypeInference.cpp +++ b/compiler/loco/src/Service/TypeInference.cpp @@ -147,6 +147,7 @@ struct CanonicalTypeForwardAlgorithm final : public loco::CanonicalNodeVisitor<l loco::DataType visit(const loco::TensorBiasAdd *node) { return loco::dtype_get(node->value()); } loco::DataType visit(const loco::TensorBroadcast *node) { return loco::dtype_get(node->input()); } loco::DataType visit(const loco::TensorSoftmax *node) { return loco::dtype_get(node->input()); } + loco::DataType visit(const loco::TransposedConv2D *node) { return loco::dtype_get(node->ifm()); } }; } // namespace |