From 92358fc122cba98e373ca2ff42b1cfe7618161e8 Mon Sep 17 00:00:00 2001 From: James Ward Date: Tue, 21 Nov 2023 18:14:43 +0000 Subject: Add Rescale Attribute changes Signed-off-by: James Ward Change-Id: Ifc5445c7129abb914a6cef952e27fccf47ea6397 --- python/serializer/tosa_serializer.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) 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 -- cgit v1.2.1