Add Rescale Attribute changes

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: Ifc5445c7129abb914a6cef952e27fccf47ea6397

1 files changed, 12 insertions, 1 deletions

diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py
index 563bc00..1b40723 100644
--- a/python/serializer/tosa_serializer.py
+++ b/python/serializer/tosa_serializer.py
@@ -283,7 +283,16 @@ class TosaSerializerAttribute(TosaSerializerUnion):
         self.floats.append((a.AddMaxFp, serialized_maxfp_bytes))
 
     def RescaleAttribute(
-        self, input_zp, output_zp, multiplier, shift, scale32, double_round, per_channel
+        self,
+        input_zp,
+        output_zp,
+        multiplier,
+        shift,
+        scale32,
+        double_round,
+        per_channel,
+        input_unsigned,
+        output_unsigned,
     ):
         from tosa import RescaleAttribute as a, Attribute
 
@@ -297,6 +306,8 @@ class TosaSerializerAttribute(TosaSerializerUnion):
         self.bools.append((a.AddScale32, scale32))
         self.bools.append((a.AddDoubleRound, double_round))
         self.bools.append((a.AddPerChannel, per_channel))
+        self.bools.append((a.AddInputUnsigned, input_unsigned))
+        self.bools.append((a.AddOutputUnsigned, output_unsigned))
 
     def MulAttribute(self, shift):
         from tosa import MulAttribute as a, Attribute