summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>2019-09-17 04:55:15 (GMT)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>2019-09-17 04:55:15 (GMT)
commit7b5c9ffd644d7d033c82100bc3afc4245ca22443 (patch)
tree94c5b90182b9d4170d4cdc493e6294afc7058b90
parentfd5551a6187f17f640bea8a0777c313115956839 (diff)
downloadnnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.zip
nnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.tar.gz
nnfw-7b5c9ffd644d7d033c82100bc3afc4245ca22443.tar.bz2
[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.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/compiler/loco/src/Service/TypeInference.cpp b/compiler/loco/src/Service/TypeInference.cpp
index 3c3c5ad..92cdbd5 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