aboutsummaryrefslogtreecommitdiff
path: root/src/TosaSerialize.cpp
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2022-06-03 18:16:40 -0700
committerEric Kunze <eric.kunze@arm.com>2022-06-13 16:25:43 -0700
commita80f98072c492f68b54f1cc0acc799e63227d1d1 (patch)
treefccab1e5cadcfab76cec6170add18235a12cd5e0 /src/TosaSerialize.cpp
parent661183cc14aa23e818f62b15e6ff1c06ef9a7e42 (diff)
downloadtosa_mlir_translator-a80f98072c492f68b54f1cc0acc799e63227d1d1.tar.gz
Remove dilation serialization for transpose_conv2d
It does not exist in the specification, removing from serialized form Signed-off-by: Eric Kunze <eric.kunze@arm.com> Change-Id: I496e1d4906b3ba1b9e00a1adef4ff3e7bab3d8ba
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r--src/TosaSerialize.cpp11
1 files changed, 2 insertions, 9 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 04dae51..60393ae 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -687,7 +687,7 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeConv2DOp>(
for (auto &int_attr : outpad_attr) {
outpad.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt());
}
- ASSERT_VECTOR_LENGTH(outpad, 2);
+ ASSERT_VECTOR_LENGTH(outpad, 4);
auto stride_attr =
op.getAttr("stride").dyn_cast<mlir::ArrayAttr>().getValue();
@@ -696,13 +696,6 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeConv2DOp>(
}
ASSERT_VECTOR_LENGTH(stride, 2);
- auto dilation_attr =
- op.getAttr("dilation").dyn_cast<mlir::ArrayAttr>().getValue();
- for (auto &int_attr : dilation_attr) {
- dilation.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt());
- }
- ASSERT_VECTOR_LENGTH(dilation, 2);
-
auto output_shape_attr =
op.getAttr("out_shape").dyn_cast<mlir::ArrayAttr>().getValue();
for (auto &int_attr : output_shape_attr) {
@@ -715,7 +708,7 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeConv2DOp>(
std::string input2_name = GetTensorName(op.getOperand(2));
std::string output_name = GetTensorName(op.getResult(0));
- TosaTransposeConvAttribute attribute(outpad, stride, dilation, output_shape);
+ TosaTransposeConvAttribute attribute(outpad, stride, output_shape);
auto quant_info =
op.getAttrOfType<mlir::tosa::ConvOpQuantizationAttr>("quantization_info");