From 485a11d8cb67c8062c632f0987cd31cedbe93d6d Mon Sep 17 00:00:00 2001 From: James Ward Date: Fri, 5 Aug 2022 13:48:37 +0100 Subject: FP16 support in serialization * Allow serialization of fp16 data * Add package to support integrated half data-type (half_float::half), independent of native float: http://half.sourceforge.net/ * Allow passing of accumulate data-type in serialization Signed-off-by: James Ward Change-Id: I54357f02e3776d81958228f699ea5044f2014f4b --- python/tosa/FullyConnectedAttribute.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'python/tosa/FullyConnectedAttribute.py') diff --git a/python/tosa/FullyConnectedAttribute.py b/python/tosa/FullyConnectedAttribute.py index 892b0da..546ec60 100644 --- a/python/tosa/FullyConnectedAttribute.py +++ b/python/tosa/FullyConnectedAttribute.py @@ -42,7 +42,14 @@ class FullyConnectedAttribute(object): return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) return 0 -def FullyConnectedAttributeStart(builder): builder.StartObject(2) + # FullyConnectedAttribute + def AccumDtype(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos) + return 0 + +def FullyConnectedAttributeStart(builder): builder.StartObject(3) def Start(builder): return FullyConnectedAttributeStart(builder) def FullyConnectedAttributeAddInputZp(builder, inputZp): builder.PrependInt32Slot(0, inputZp, 0) @@ -51,6 +58,9 @@ def AddInputZp(builder, inputZp): def FullyConnectedAttributeAddWeightZp(builder, weightZp): builder.PrependInt32Slot(1, weightZp, 0) def AddWeightZp(builder, weightZp): return FullyConnectedAttributeAddWeightZp(builder, weightZp) +def FullyConnectedAttributeAddAccumDtype(builder, accumDtype): builder.PrependUint32Slot(2, accumDtype, 0) +def AddAccumDtype(builder, accumDtype): + return FullyConnectedAttributeAddAccumDtype(builder, accumDtype) def FullyConnectedAttributeEnd(builder): return builder.EndObject() def End(builder): return FullyConnectedAttributeEnd(builder) \ No newline at end of file -- cgit v1.2.1