diff options
author | Rob Elliott <robert.elliott@arm.com> | 2023-08-17 14:27:06 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-08-21 16:14:51 +0000 |
commit | 00a15db3e1a188b25065d095152d701f4394cdc5 (patch) | |
tree | 96761b9f7ac3ad759f9f0ffbf63a6d0ef115ad14 /ethosu/vela/tosa/SliceAttribute.py | |
parent | 8ea90edb75e5d2353aa91c264356fc9d460ca308 (diff) | |
download | ethos-u-vela-00a15db3e1a188b25065d095152d701f4394cdc5.tar.gz |
Moving Vela to use TOSA v0.80.0 specification
* Using serialization_lib main branch to update statically copied
files sha 5f920211ac23393a7b98a0d358bfbfc3232d5c8f (v0.80.0)
* All files within the ethosu/vela/tosa are copied from that revision
* Note: hope to move to serialization_lib as a pip module in future
* Modified the ethosu/vela/{tosa_mapping,tosa_reader}.py to use
v0.80.0 TOSA FlatBuffers implementation
* These are the additional changes made to support this new version,
with changes in the format of the FlatBuffers file and where various
values are stored. Either changing from input to attribute, or
moving to different attributes.
Signed-off-by: Rob Elliott <robert.elliott@arm.com>
Change-Id: I5e1fcc2a9964148619be3477adf1e88e84cbae2d
Diffstat (limited to 'ethosu/vela/tosa/SliceAttribute.py')
-rw-r--r-- | ethosu/vela/tosa/SliceAttribute.py | 69 |
1 files changed, 59 insertions, 10 deletions
diff --git a/ethosu/vela/tosa/SliceAttribute.py b/ethosu/vela/tosa/SliceAttribute.py index d2f99584..6d8e00bb 100644 --- a/ethosu/vela/tosa/SliceAttribute.py +++ b/ethosu/vela/tosa/SliceAttribute.py @@ -3,23 +3,33 @@ # namespace: tosa import flatbuffers +from flatbuffers.compat import import_numpy +np = import_numpy() class SliceAttribute(object): __slots__ = ['_tab'] @classmethod - def GetRootAsSliceAttribute(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = SliceAttribute() x.Init(buf, n + offset) return x + @classmethod + def GetRootAsSliceAttribute(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod + def SliceAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False): + return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed) + # SliceAttribute def Init(self, buf, pos): self._tab = flatbuffers.table.Table(buf, pos) # SliceAttribute - def Begin(self, j): + def Start(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: a = self._tab.Vector(o) @@ -27,20 +37,25 @@ class SliceAttribute(object): return 0 # SliceAttribute - def BeginAsNumpy(self): + def StartAsNumpy(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o) return 0 # SliceAttribute - def BeginLength(self): + def StartLength(self): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: return self._tab.VectorLen(o) return 0 # SliceAttribute + def StartIsNone(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) + return o == 0 + + # SliceAttribute def Size(self, j): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) if o != 0: @@ -62,9 +77,43 @@ class SliceAttribute(object): return self._tab.VectorLen(o) return 0 -def SliceAttributeStart(builder): builder.StartObject(2) -def SliceAttributeAddBegin(builder, begin): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(begin), 0) -def SliceAttributeStartBeginVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def SliceAttributeAddSize(builder, size): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(size), 0) -def SliceAttributeStartSizeVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def SliceAttributeEnd(builder): return builder.EndObject() + # SliceAttribute + def SizeIsNone(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) + return o == 0 + +def SliceAttributeStart(builder): + builder.StartObject(2) + +def Start(builder): + SliceAttributeStart(builder) + +def SliceAttributeAddStart(builder, start): + builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(start), 0) + +def AddStart(builder, start): + SliceAttributeAddStart(builder, start) + +def SliceAttributeStartStartVector(builder, numElems): + return builder.StartVector(4, numElems, 4) + +def StartStartVector(builder, numElems: int) -> int: + return SliceAttributeStartStartVector(builder, numElems) + +def SliceAttributeAddSize(builder, size): + builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(size), 0) + +def AddSize(builder, size): + SliceAttributeAddSize(builder, size) + +def SliceAttributeStartSizeVector(builder, numElems): + return builder.StartVector(4, numElems, 4) + +def StartSizeVector(builder, numElems: int) -> int: + return SliceAttributeStartSizeVector(builder, numElems) + +def SliceAttributeEnd(builder): + return builder.EndObject() + +def End(builder): + return SliceAttributeEnd(builder) |