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/ConvAttribute.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) (limited to 'python/tosa/ConvAttribute.py') diff --git a/python/tosa/ConvAttribute.py b/python/tosa/ConvAttribute.py index 8244ea5..fb22c7a 100644 --- a/python/tosa/ConvAttribute.py +++ b/python/tosa/ConvAttribute.py @@ -10,13 +10,17 @@ class ConvAttribute(object): __slots__ = ['_tab'] @classmethod - def GetRootAsConvAttribute(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = ConvAttribute() x.Init(buf, n + offset) return x @classmethod + def GetRootAsConvAttribute(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod def ConvAttributeBufferHasIdentifier(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 ConvAttribute(object): return 0 def ConvAttributeStart(builder): builder.StartObject(5) +def Start(builder): + return ConvAttributeStart(builder) def ConvAttributeAddPad(builder, pad): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(pad), 0) +def AddPad(builder, pad): + return ConvAttributeAddPad(builder, pad) def ConvAttributeStartPadVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartPadVector(builder, numElems): + return ConvAttributeStartPadVector(builder, numElems) def ConvAttributeAddStride(builder, stride): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(stride), 0) +def AddStride(builder, stride): + return ConvAttributeAddStride(builder, stride) def ConvAttributeStartStrideVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartStrideVector(builder, numElems): + return ConvAttributeStartStrideVector(builder, numElems) def ConvAttributeAddDilation(builder, dilation): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(dilation), 0) +def AddDilation(builder, dilation): + return ConvAttributeAddDilation(builder, dilation) def ConvAttributeStartDilationVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def StartDilationVector(builder, numElems): + return ConvAttributeStartDilationVector(builder, numElems) def ConvAttributeAddInputZp(builder, inputZp): builder.PrependInt32Slot(3, inputZp, 0) +def AddInputZp(builder, inputZp): + return ConvAttributeAddInputZp(builder, inputZp) def ConvAttributeAddWeightZp(builder, weightZp): builder.PrependInt32Slot(4, weightZp, 0) +def AddWeightZp(builder, weightZp): + return ConvAttributeAddWeightZp(builder, weightZp) def ConvAttributeEnd(builder): return builder.EndObject() +def End(builder): + return ConvAttributeEnd(builder) \ No newline at end of file -- cgit v1.2.1