diff options
author | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-03-14 23:43:29 +0000 |
---|---|---|
committer | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-04-18 21:41:58 -0700 |
commit | 793a6c5088d1cb59753eccc158173fbab8e44190 (patch) | |
tree | 17d1cba9a70b42fcdc719c42ca03463b90f1a87a /src/TosaSerialize.cpp | |
parent | 6103155b3a2a555c3fc4a3a2173b35ea573c9600 (diff) | |
download | tosa_mlir_translator-793a6c5088d1cb59753eccc158173fbab8e44190.tar.gz |
[tosa-mlir-translator] Remove TRANSPOSE_CONV2D out_shape argument
Change-Id: I0a96662eed0466cc2220f88522fb97d3ad221559
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index 6553944..c3a9878 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -888,9 +888,6 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeConv2DOp>( auto stride = getDenseI64ArrayAttr<int>(op.getAttr("stride")); ASSERT_VECTOR_LENGTH(stride, 2); - auto out_shape = getDenseI64ArrayAttr<int>(op.getAttr("out_shape")); - ASSERT_VECTOR_LENGTH(out_shape, 4); - std::string input0_name = GetTensorName(op.getOperand(0)); std::string input1_name = GetTensorName(op.getOperand(1)); std::string input2_name = GetTensorName(op.getOperand(2)); @@ -916,8 +913,8 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeConv2DOp>( auto acc_type = op.getAttr("acc_type").cast<mlir::TypeAttr>().getValue(); auto acc_dtype = Type2AccDType(acc_type); - TosaTransposeConvAttribute attribute(out_pad, stride, out_shape, input_zp, - weight_zp, local_bound, acc_dtype); + TosaTransposeConvAttribute attribute(out_pad, stride, input_zp, weight_zp, + local_bound, acc_dtype); TosaSerializationOperator *tyop = new TosaSerializationOperator( Op_TRANSPOSE_CONV2D, Attribute_TransposeConvAttribute, &attribute, |