From fc32f56a067c526238c15de097fe78fdcab95cb5 Mon Sep 17 00:00:00 2001
From: James Ward <james.ward@arm.com>
Date: Tue, 14 Nov 2023 18:45:33 +0000
Subject: Add Rescale Attribute changes

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I8ac71800d922526aad0a7c351ad1943481208cc2
---
 src/TosaSerialize.cpp | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 9807a99..263f51c 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1415,6 +1415,11 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RescaleOp>(
       op.getAttr("multiplier").dyn_cast<mlir::DenseI32ArrayAttr>().asArrayRef();
   auto shift = getDenseI8ArrayAttr<int32_t>(op.getAttr("shift"));
 
+  bool input_unsigned =
+      op.getAttr("input_unsigned").dyn_cast<mlir::BoolAttr>().getValue();
+  bool output_unsigned =
+      op.getAttr("output_unsigned").dyn_cast<mlir::BoolAttr>().getValue();
+
   auto input = op.getOperand(0);
   auto input_ty = input.getType().cast<mlir::RankedTensorType>();
   auto output = op.getResult(0);
@@ -1423,9 +1428,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RescaleOp>(
   std::string input_name = GetTensorName(input);
   std::string output_name = GetTensorName(output);
 
-  TosaRescaleAttribute attribute(
-      input_zp, output_zp, multiplier, shift, scale32, double_round,
-      per_channel, input_ty.isUnsignedInteger(), output_ty.isUnsignedInteger());
+  TosaRescaleAttribute attribute(input_zp, output_zp, multiplier, shift,
+                                 scale32, double_round, per_channel,
+                                 input_unsigned, output_unsigned);
 
   TosaSerializationOperator *tyop = new TosaSerializationOperator(
       Op_RESCALE, Attribute_RescaleAttribute, &attribute,
-- 
cgit v1.2.1