From 4381b3d7fcb7cab975f46c62c86a35c53ade047f Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Mon, 22 Aug 2022 18:15:41 +0000 Subject: Update serialization lib to flatbuffers 2.0.6 Signed-off-by: Eric Kunze Change-Id: I64ac03a2fa8711c2050efd5fedbc34a4c2a67bd5 --- python/tosa/TosaTensor.py | 22 +++++++++++++++++++++- 1 file changed, 21 insertions(+), 1 deletion(-) (limited to 'python/tosa/TosaTensor.py') diff --git a/python/tosa/TosaTensor.py b/python/tosa/TosaTensor.py index a6f609d..d8ee267 100644 --- a/python/tosa/TosaTensor.py +++ b/python/tosa/TosaTensor.py @@ -10,13 +10,17 @@ class TosaTensor(object): __slots__ = ['_tab'] @classmethod - def GetRootAsTosaTensor(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = TosaTensor() x.Init(buf, n + offset) return x @classmethod + def GetRootAsTosaTensor(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod def TosaTensorBufferHasIdentifier(cls, buf, offset, size_prefixed=False): return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed) @@ -93,10 +97,26 @@ class TosaTensor(object): return o == 0 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) +def AddName(builder, name): + return 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): + return TosaTensorStartShapeVector(builder, numElems) 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) +def AddData(builder, data): + return TosaTensorAddData(builder, data) def TosaTensorStartDataVector(builder, numElems): return builder.StartVector(1, numElems, 1) +def StartDataVector(builder, numElems): + return TosaTensorStartDataVector(builder, numElems) def TosaTensorEnd(builder): return builder.EndObject() +def End(builder): + return TosaTensorEnd(builder) \ No newline at end of file -- cgit v1.2.1