From 89963aa8fad822ab7a6e1ff92f6b7b4ee0b9350c Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Mon, 3 Jul 2023 22:14:05 +0000 Subject: 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 Change-Id: I7278dfe9c7b55b9afbe2329e3058b63a1cdce586 --- python/tosa/TosaTensor.py | 55 +++++++++++++++++++++++++++++++++-------------- 1 file changed, 39 insertions(+), 16 deletions(-) (limited to 'python/tosa/TosaTensor.py') 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) -- cgit v1.2.1