diff options
author | Eric Kunze <eric.kunze@arm.com> | 2023-11-01 16:12:07 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-01 16:14:25 -0700 |
commit | 4881c29247d4b411de446b13d9bd58ea93737aac (patch) | |
tree | c1a55620543fd0fff382a9f8c7ceeab14b6aa5db /python/tosa/RFFTAttribute.py | |
parent | 5917fc7a9392da8fd1e8c68b2d00b89709a31584 (diff) | |
download | serialization_lib-4881c29247d4b411de446b13d9bd58ea93737aac.tar.gz |
Add support for local_bound attribute
local_bound is used to determine when fast convolution algorithms
can be used in implementing the operation.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I9970a2544e90a620f46ac4d3d01cec90a15710a9
Diffstat (limited to 'python/tosa/RFFTAttribute.py')
-rw-r--r-- | python/tosa/RFFTAttribute.py | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/python/tosa/RFFTAttribute.py b/python/tosa/RFFTAttribute.py new file mode 100644 index 0000000..7f76024 --- /dev/null +++ b/python/tosa/RFFTAttribute.py @@ -0,0 +1,54 @@ +# automatically generated by the FlatBuffers compiler, do not modify + +# namespace: tosa + +import flatbuffers +from flatbuffers.compat import import_numpy +np = import_numpy() + +class RFFTAttribute(object): + __slots__ = ['_tab'] + + @classmethod + def GetRootAs(cls, buf, offset=0): + n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) + x = RFFTAttribute() + x.Init(buf, n + offset) + return x + + @classmethod + def GetRootAsRFFTAttribute(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod + def RFFTAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False): + return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed) + + # RFFTAttribute + def Init(self, buf, pos): + self._tab = flatbuffers.table.Table(buf, pos) + + # RFFTAttribute + def LocalBound(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) + if o != 0: + return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) + return False + +def RFFTAttributeStart(builder): + builder.StartObject(1) + +def Start(builder): + RFFTAttributeStart(builder) + +def RFFTAttributeAddLocalBound(builder, localBound): + builder.PrependBoolSlot(0, localBound, 0) + +def AddLocalBound(builder, localBound): + RFFTAttributeAddLocalBound(builder, localBound) + +def RFFTAttributeEnd(builder): + return builder.EndObject() + +def End(builder): + return RFFTAttributeEnd(builder) |