commit 793a6c5088d1cb59753eccc158173fbab8e44190
author Suraj Sudhir <suraj.sudhir@arm.com> Thu Mar 14 23:43:29 2024 +0000
committer Suraj Sudhir <suraj.sudhir@arm.com> Thu Apr 18 21:41:58 2024 -0700
tree 17d1cba9a70b42fcdc719c42ca03463b90f1a87a
parent 6103155b3a2a555c3fc4a3a2173b35ea573c9600

[tosa-mlir-translator] Remove TRANSPOSE_CONV2D out_shape argument

Change-Id: I0a96662eed0466cc2220f88522fb97d3ad221559
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>

src/TosaDeserialize.cpp

diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp
index fdbd892..215d760 100644
--- a/src/TosaDeserialize.cpp
+++ b/src/TosaDeserialize.cpp
@@ -842,8 +842,6 @@
       BuildDenseI64ArrayAttr(op_builder, attr->out_pad());
   mlir::DenseI64ArrayAttr stride =
       BuildDenseI64ArrayAttr(op_builder, attr->stride());
-  mlir::DenseI64ArrayAttr output_shape =
-      BuildDenseI64ArrayAttr(op_builder, attr->output_shape());
   auto input_zp = attr->input_zp();
   auto weight_zp = attr->weight_zp();
   bool local_bound = attr->local_bound();
@@ -860,7 +858,7 @@
 
   mlir_op = op_builder->create<mlir::tosa::TransposeConv2DOp>(
       loc, output_type, input0_val, input1_val, input2_val, out_pad, stride,
-      output_shape, acc_type, input_zp_attr, weight_zp_attr, local_bound);
+      acc_type, input_zp_attr, weight_zp_attr, local_bound);
 
   block->push_back(mlir_op);
   return std::vector<mlir::Value>({mlir_op->getResult(0)});

src/TosaSerialize.cpp

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 @@
   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 @@
   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,

third_party/serialization_lib

diff --git a/third_party/serialization_lib b/third_party/serialization_lib
index 57d7818..50256e1 160000
--- a/third_party/serialization_lib
+++ b/third_party/serialization_lib
@@ -1 +1 @@
-Subproject commit 57d781883142db8a45fe98ac1a1dfacc49cba78a
+Subproject commit 50256e168c3e759f03445bb872d0a43da1a6ba30