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/PoolAttribute.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) (limited to 'python/tosa/PoolAttribute.py') diff --git a/python/tosa/PoolAttribute.py b/python/tosa/PoolAttribute.py index 8b6903e..8256a6d 100644 --- a/python/tosa/PoolAttribute.py +++ b/python/tosa/PoolAttribute.py @@ -10,13 +10,17 @@ class PoolAttribute(object): __slots__ = ['_tab'] @classmethod - def GetRootAsPoolAttribute(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = PoolAttribute() x.Init(buf, n + offset) return x @classmethod + def GetRootAsPoolAttribute(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod def PoolAttributeBufferHasIdentifier(cls, buf, offset, size_prefixed=False): return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x4F\x53\x41", size_prefixed=size_prefixed) @@ -120,12 +124,32 @@ class PoolAttribute(object): return 0 def PoolAttributeStart(builder): builder.StartObject(5) +def Start(builder): + return PoolAttributeStart(builder) def PoolAttributeAddPad(builder, pad): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(pad), 0) +def AddPad(builder, pad): + return PoolAttributeAddPad(builder, pad) def PoolAttributeStartPadVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartPadVector(builder, numElems): + return PoolAttributeStartPadVector(builder, numElems) def PoolAttributeAddKernel(builder, kernel): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(kernel), 0) +def AddKernel(builder, kernel): + return PoolAttributeAddKernel(builder, kernel) def PoolAttributeStartKernelVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartKernelVector(builder, numElems): + return PoolAttributeStartKernelVector(builder, numElems) def PoolAttributeAddStride(builder, stride): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(stride), 0) +def AddStride(builder, stride): + return PoolAttributeAddStride(builder, stride) def PoolAttributeStartStrideVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartStrideVector(builder, numElems): + return PoolAttributeStartStrideVector(builder, numElems) def PoolAttributeAddInputZp(builder, inputZp): builder.PrependInt32Slot(3, inputZp, 0) +def AddInputZp(builder, inputZp): + return PoolAttributeAddInputZp(builder, inputZp) def PoolAttributeAddOutputZp(builder, outputZp): builder.PrependInt32Slot(4, outputZp, 0) +def AddOutputZp(builder, outputZp): + return PoolAttributeAddOutputZp(builder, outputZp) def PoolAttributeEnd(builder): return builder.EndObject() +def End(builder): + return PoolAttributeEnd(builder) \ No newline at end of file -- cgit v1.2.1