diff options
author | James Ward <james.ward@arm.com> | 2023-11-21 18:14:43 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-29 23:30:49 +0000 |
commit | 0925d40d0cf64ea1836218671f29a2c37df13662 (patch) | |
tree | 589d29b8c4b6482c161f56fc4e976509ab2c9b8b | |
parent | cce787e55f2f4e9fa14fd4643ddfd34de754c500 (diff) | |
download | serialization_lib-0925d40d0cf64ea1836218671f29a2c37df13662.tar.gz |
Add Rescale Attribute changes
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: Ifc5445c7129abb914a6cef952e27fccf47ea6397
-rw-r--r-- | python/serializer/tosa_serializer.py | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py index 0a3fff6..da7c62c 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 |