From bdcc3fee1b8bf55aac50e060115b92a1ccf9741c Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Tue, 7 Jun 2022 05:17:37 +0000 Subject: Remove quantinfo types Any needed information has been moved into the attributes for each operator. This aligns with the structure of the attributes in the TOSA specification, and generally simplifies the code. Change-Id: I8243e91b09de1a9115f8af09c5e7def7e8f2866b Signed-off-by: Eric Kunze --- python/tosa/MatMulAttribute.py | 44 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 python/tosa/MatMulAttribute.py (limited to 'python/tosa/MatMulAttribute.py') diff --git a/python/tosa/MatMulAttribute.py b/python/tosa/MatMulAttribute.py new file mode 100644 index 0000000..601f13f --- /dev/null +++ b/python/tosa/MatMulAttribute.py @@ -0,0 +1,44 @@ +# automatically generated by the FlatBuffers compiler, do not modify + +# namespace: tosa + +import flatbuffers +from flatbuffers.compat import import_numpy +np = import_numpy() + +class MatMulAttribute(object): + __slots__ = ['_tab'] + + @classmethod + def GetRootAsMatMulAttribute(cls, buf, offset): + n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) + x = MatMulAttribute() + x.Init(buf, n + offset) + return x + + @classmethod + def MatMulAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False): + return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed) + + # MatMulAttribute + def Init(self, buf, pos): + self._tab = flatbuffers.table.Table(buf, pos) + + # MatMulAttribute + def AZp(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) + return 0 + + # MatMulAttribute + def BZp(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos) + return 0 + +def MatMulAttributeStart(builder): builder.StartObject(2) +def MatMulAttributeAddAZp(builder, aZp): builder.PrependInt32Slot(0, aZp, 0) +def MatMulAttributeAddBZp(builder, bZp): builder.PrependInt32Slot(1, bZp, 0) +def MatMulAttributeEnd(builder): return builder.EndObject() -- cgit v1.2.1