Add Rescale Attribute changes
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: Ifc5445c7129abb914a6cef952e27fccf47ea6397
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 @@
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 @@
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