From 793a6c5088d1cb59753eccc158173fbab8e44190 Mon Sep 17 00:00:00 2001
From: Suraj Sudhir <suraj.sudhir@arm.com>
Date: Thu, 14 Mar 2024 23:43:29 +0000
Subject: [tosa-mlir-translator] Remove TRANSPOSE_CONV2D out_shape argument

Change-Id: I0a96662eed0466cc2220f88522fb97d3ad221559
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
---
 src/TosaSerialize.cpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

(limited to '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 @@ 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,
-- 
cgit v1.2.1