diff options
author | Tai Ly <tai.ly@arm.com> | 2023-07-03 22:14:05 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2023-07-18 21:00:58 +0000 |
commit | 89963aa8fad822ab7a6e1ff92f6b7b4ee0b9350c (patch) | |
tree | 123b51806c246b5c10fe20d2f90609d03d373b0c /python/tosa/TosaTensor.py | |
parent | 0bc46e44e6501788c4661d1f03572c14ea2ecc89 (diff) | |
download | serialization_lib-89963aa8fad822ab7a6e1ff92f6b7b4ee0b9350c.tar.gz |
Update to use GenText
changed GenerateText to GenText, with reversed return polarity.
updated flatbuffer to v23.5.26, and regenerated headers
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I7278dfe9c7b55b9afbe2329e3058b63a1cdce586
Diffstat (limited to 'python/tosa/TosaTensor.py')
-rw-r--r-- | python/tosa/TosaTensor.py | 55 |
1 files changed, 39 insertions, 16 deletions
diff --git a/python/tosa/TosaTensor.py b/python/tosa/TosaTensor.py index d8ee267..850ff8f 100644 --- a/python/tosa/TosaTensor.py +++ b/python/tosa/TosaTensor.py @@ -96,27 +96,50 @@ class TosaTensor(object): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) return o == 0 -def TosaTensorStart(builder): builder.StartObject(4) +def TosaTensorStart(builder): + builder.StartObject(4) + def Start(builder): - return TosaTensorStart(builder) -def TosaTensorAddName(builder, name): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0) + TosaTensorStart(builder) + +def TosaTensorAddName(builder, name): + builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0) + def AddName(builder, name): - return TosaTensorAddName(builder, name) -def TosaTensorAddShape(builder, shape): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) + TosaTensorAddName(builder, name) + +def TosaTensorAddShape(builder, shape): + builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0) + def AddShape(builder, shape): - return TosaTensorAddShape(builder, shape) -def TosaTensorStartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def StartShapeVector(builder, numElems): + TosaTensorAddShape(builder, shape) + +def TosaTensorStartShapeVector(builder, numElems): + return builder.StartVector(4, numElems, 4) + +def StartShapeVector(builder, numElems: int) -> int: return TosaTensorStartShapeVector(builder, numElems) -def TosaTensorAddType(builder, type): builder.PrependUint32Slot(2, type, 0) + +def TosaTensorAddType(builder, type): + builder.PrependUint32Slot(2, type, 0) + def AddType(builder, type): - return TosaTensorAddType(builder, type) -def TosaTensorAddData(builder, data): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(data), 0) + TosaTensorAddType(builder, type) + +def TosaTensorAddData(builder, data): + builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(data), 0) + def AddData(builder, data): - return TosaTensorAddData(builder, data) -def TosaTensorStartDataVector(builder, numElems): return builder.StartVector(1, numElems, 1) -def StartDataVector(builder, numElems): + TosaTensorAddData(builder, data) + +def TosaTensorStartDataVector(builder, numElems): + return builder.StartVector(1, numElems, 1) + +def StartDataVector(builder, numElems: int) -> int: return TosaTensorStartDataVector(builder, numElems) -def TosaTensorEnd(builder): return builder.EndObject() + +def TosaTensorEnd(builder): + return builder.EndObject() + def End(builder): - return TosaTensorEnd(builder)
\ No newline at end of file + return TosaTensorEnd(builder) |