diff options
Diffstat (limited to 'ethosu/vela/tflite/Conv2DOptions.py')
-rw-r--r-- | ethosu/vela/tflite/Conv2DOptions.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/Conv2DOptions.py b/ethosu/vela/tflite/Conv2DOptions.py index a05b39c..5928009 100644 --- a/ethosu/vela/tflite/Conv2DOptions.py +++ b/ethosu/vela/tflite/Conv2DOptions.py @@ -3,18 +3,24 @@ # namespace: tflite import flatbuffers +from flatbuffers.compat import import_numpy +np = import_numpy() class Conv2DOptions(object): __slots__ = ['_tab'] @classmethod - def GetRootAsConv2DOptions(cls, buf, offset): + def GetRootAs(cls, buf, offset=0): n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) x = Conv2DOptions() x.Init(buf, n + offset) return x @classmethod + def GetRootAsConv2DOptions(cls, buf, offset=0): + """This method is deprecated. Please switch to GetRootAs.""" + return cls.GetRootAs(buf, offset) + @classmethod def Conv2DOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False): return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed) @@ -65,10 +71,26 @@ class Conv2DOptions(object): return 1 def Conv2DOptionsStart(builder): builder.StartObject(6) +def Start(builder): + return Conv2DOptionsStart(builder) def Conv2DOptionsAddPadding(builder, padding): builder.PrependInt8Slot(0, padding, 0) +def AddPadding(builder, padding): + return Conv2DOptionsAddPadding(builder, padding) def Conv2DOptionsAddStrideW(builder, strideW): builder.PrependInt32Slot(1, strideW, 0) +def AddStrideW(builder, strideW): + return Conv2DOptionsAddStrideW(builder, strideW) def Conv2DOptionsAddStrideH(builder, strideH): builder.PrependInt32Slot(2, strideH, 0) +def AddStrideH(builder, strideH): + return Conv2DOptionsAddStrideH(builder, strideH) def Conv2DOptionsAddFusedActivationFunction(builder, fusedActivationFunction): builder.PrependInt8Slot(3, fusedActivationFunction, 0) +def AddFusedActivationFunction(builder, fusedActivationFunction): + return Conv2DOptionsAddFusedActivationFunction(builder, fusedActivationFunction) def Conv2DOptionsAddDilationWFactor(builder, dilationWFactor): builder.PrependInt32Slot(4, dilationWFactor, 1) +def AddDilationWFactor(builder, dilationWFactor): + return Conv2DOptionsAddDilationWFactor(builder, dilationWFactor) def Conv2DOptionsAddDilationHFactor(builder, dilationHFactor): builder.PrependInt32Slot(5, dilationHFactor, 1) +def AddDilationHFactor(builder, dilationHFactor): + return Conv2DOptionsAddDilationHFactor(builder, dilationHFactor) def Conv2DOptionsEnd(builder): return builder.EndObject() +def End(builder): + return Conv2DOptionsEnd(builder)
\ No newline at end of file |