diff options
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 11 |
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"); |