| # automatically generated by the FlatBuffers compiler, do not modify |
| |
| # namespace: tflite |
| |
| import flatbuffers |
| from flatbuffers.compat import import_numpy |
| np = import_numpy() |
| |
| class StablehloConvolutionOptions(object): |
| __slots__ = ['_tab'] |
| |
| @classmethod |
| def GetRootAs(cls, buf, offset=0): |
| n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset) |
| x = StablehloConvolutionOptions() |
| x.Init(buf, n + offset) |
| return x |
| |
| @classmethod |
| def GetRootAsStablehloConvolutionOptions(cls, buf, offset=0): |
| """This method is deprecated. Please switch to GetRootAs.""" |
| return cls.GetRootAs(buf, offset) |
| @classmethod |
| def StablehloConvolutionOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False): |
| return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed) |
| |
| # StablehloConvolutionOptions |
| def Init(self, buf, pos): |
| self._tab = flatbuffers.table.Table(buf, pos) |
| |
| # StablehloConvolutionOptions |
| def WindowStrides(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowStridesAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowStridesLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowStridesIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def Padding(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PaddingAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PaddingLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PaddingIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def LhsDilation(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def LhsDilationAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def LhsDilationLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def LhsDilationIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def RhsDilation(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def RhsDilationAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def RhsDilationLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def RhsDilationIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def WindowReversal(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.BoolFlags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowReversalAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.BoolFlags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowReversalLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def WindowReversalIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def InputBatchDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def InputFeatureDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def InputSpatialDimensions(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def InputSpatialDimensionsAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def InputSpatialDimensionsLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def InputSpatialDimensionsIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def KernelInputFeatureDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def KernelOutputFeatureDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def KernelSpatialDimensions(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def KernelSpatialDimensionsAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def KernelSpatialDimensionsLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def KernelSpatialDimensionsIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def OutputBatchDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def OutputFeatureDimension(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(28)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def OutputSpatialDimensions(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(30)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def OutputSpatialDimensionsAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(30)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def OutputSpatialDimensionsLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(30)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def OutputSpatialDimensionsIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(30)) |
| return o == 0 |
| |
| # StablehloConvolutionOptions |
| def FeatureGroupCount(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(32)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def BatchGroupCount(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(34)) |
| if o != 0: |
| return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PrecisionConfig(self, j): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(36)) |
| if o != 0: |
| a = self._tab.Vector(o) |
| return self._tab.Get(flatbuffers.number_types.Uint32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4)) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PrecisionConfigAsNumpy(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(36)) |
| if o != 0: |
| return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint32Flags, o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PrecisionConfigLength(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(36)) |
| if o != 0: |
| return self._tab.VectorLen(o) |
| return 0 |
| |
| # StablehloConvolutionOptions |
| def PrecisionConfigIsNone(self): |
| o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(36)) |
| return o == 0 |
| |
| def StablehloConvolutionOptionsStart(builder): builder.StartObject(17) |
| def Start(builder): |
| return StablehloConvolutionOptionsStart(builder) |
| def StablehloConvolutionOptionsAddWindowStrides(builder, windowStrides): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(windowStrides), 0) |
| def AddWindowStrides(builder, windowStrides): |
| return StablehloConvolutionOptionsAddWindowStrides(builder, windowStrides) |
| def StablehloConvolutionOptionsStartWindowStridesVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartWindowStridesVector(builder, numElems): |
| return StablehloConvolutionOptionsStartWindowStridesVector(builder, numElems) |
| def StablehloConvolutionOptionsAddPadding(builder, padding): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(padding), 0) |
| def AddPadding(builder, padding): |
| return StablehloConvolutionOptionsAddPadding(builder, padding) |
| def StablehloConvolutionOptionsStartPaddingVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartPaddingVector(builder, numElems): |
| return StablehloConvolutionOptionsStartPaddingVector(builder, numElems) |
| def StablehloConvolutionOptionsAddLhsDilation(builder, lhsDilation): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(lhsDilation), 0) |
| def AddLhsDilation(builder, lhsDilation): |
| return StablehloConvolutionOptionsAddLhsDilation(builder, lhsDilation) |
| def StablehloConvolutionOptionsStartLhsDilationVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartLhsDilationVector(builder, numElems): |
| return StablehloConvolutionOptionsStartLhsDilationVector(builder, numElems) |
| def StablehloConvolutionOptionsAddRhsDilation(builder, rhsDilation): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(rhsDilation), 0) |
| def AddRhsDilation(builder, rhsDilation): |
| return StablehloConvolutionOptionsAddRhsDilation(builder, rhsDilation) |
| def StablehloConvolutionOptionsStartRhsDilationVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartRhsDilationVector(builder, numElems): |
| return StablehloConvolutionOptionsStartRhsDilationVector(builder, numElems) |
| def StablehloConvolutionOptionsAddWindowReversal(builder, windowReversal): builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(windowReversal), 0) |
| def AddWindowReversal(builder, windowReversal): |
| return StablehloConvolutionOptionsAddWindowReversal(builder, windowReversal) |
| def StablehloConvolutionOptionsStartWindowReversalVector(builder, numElems): return builder.StartVector(1, numElems, 1) |
| def StartWindowReversalVector(builder, numElems): |
| return StablehloConvolutionOptionsStartWindowReversalVector(builder, numElems) |
| def StablehloConvolutionOptionsAddInputBatchDimension(builder, inputBatchDimension): builder.PrependInt64Slot(5, inputBatchDimension, 0) |
| def AddInputBatchDimension(builder, inputBatchDimension): |
| return StablehloConvolutionOptionsAddInputBatchDimension(builder, inputBatchDimension) |
| def StablehloConvolutionOptionsAddInputFeatureDimension(builder, inputFeatureDimension): builder.PrependInt64Slot(6, inputFeatureDimension, 0) |
| def AddInputFeatureDimension(builder, inputFeatureDimension): |
| return StablehloConvolutionOptionsAddInputFeatureDimension(builder, inputFeatureDimension) |
| def StablehloConvolutionOptionsAddInputSpatialDimensions(builder, inputSpatialDimensions): builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(inputSpatialDimensions), 0) |
| def AddInputSpatialDimensions(builder, inputSpatialDimensions): |
| return StablehloConvolutionOptionsAddInputSpatialDimensions(builder, inputSpatialDimensions) |
| def StablehloConvolutionOptionsStartInputSpatialDimensionsVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartInputSpatialDimensionsVector(builder, numElems): |
| return StablehloConvolutionOptionsStartInputSpatialDimensionsVector(builder, numElems) |
| def StablehloConvolutionOptionsAddKernelInputFeatureDimension(builder, kernelInputFeatureDimension): builder.PrependInt64Slot(8, kernelInputFeatureDimension, 0) |
| def AddKernelInputFeatureDimension(builder, kernelInputFeatureDimension): |
| return StablehloConvolutionOptionsAddKernelInputFeatureDimension(builder, kernelInputFeatureDimension) |
| def StablehloConvolutionOptionsAddKernelOutputFeatureDimension(builder, kernelOutputFeatureDimension): builder.PrependInt64Slot(9, kernelOutputFeatureDimension, 0) |
| def AddKernelOutputFeatureDimension(builder, kernelOutputFeatureDimension): |
| return StablehloConvolutionOptionsAddKernelOutputFeatureDimension(builder, kernelOutputFeatureDimension) |
| def StablehloConvolutionOptionsAddKernelSpatialDimensions(builder, kernelSpatialDimensions): builder.PrependUOffsetTRelativeSlot(10, flatbuffers.number_types.UOffsetTFlags.py_type(kernelSpatialDimensions), 0) |
| def AddKernelSpatialDimensions(builder, kernelSpatialDimensions): |
| return StablehloConvolutionOptionsAddKernelSpatialDimensions(builder, kernelSpatialDimensions) |
| def StablehloConvolutionOptionsStartKernelSpatialDimensionsVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartKernelSpatialDimensionsVector(builder, numElems): |
| return StablehloConvolutionOptionsStartKernelSpatialDimensionsVector(builder, numElems) |
| def StablehloConvolutionOptionsAddOutputBatchDimension(builder, outputBatchDimension): builder.PrependInt64Slot(11, outputBatchDimension, 0) |
| def AddOutputBatchDimension(builder, outputBatchDimension): |
| return StablehloConvolutionOptionsAddOutputBatchDimension(builder, outputBatchDimension) |
| def StablehloConvolutionOptionsAddOutputFeatureDimension(builder, outputFeatureDimension): builder.PrependInt64Slot(12, outputFeatureDimension, 0) |
| def AddOutputFeatureDimension(builder, outputFeatureDimension): |
| return StablehloConvolutionOptionsAddOutputFeatureDimension(builder, outputFeatureDimension) |
| def StablehloConvolutionOptionsAddOutputSpatialDimensions(builder, outputSpatialDimensions): builder.PrependUOffsetTRelativeSlot(13, flatbuffers.number_types.UOffsetTFlags.py_type(outputSpatialDimensions), 0) |
| def AddOutputSpatialDimensions(builder, outputSpatialDimensions): |
| return StablehloConvolutionOptionsAddOutputSpatialDimensions(builder, outputSpatialDimensions) |
| def StablehloConvolutionOptionsStartOutputSpatialDimensionsVector(builder, numElems): return builder.StartVector(8, numElems, 8) |
| def StartOutputSpatialDimensionsVector(builder, numElems): |
| return StablehloConvolutionOptionsStartOutputSpatialDimensionsVector(builder, numElems) |
| def StablehloConvolutionOptionsAddFeatureGroupCount(builder, featureGroupCount): builder.PrependInt64Slot(14, featureGroupCount, 0) |
| def AddFeatureGroupCount(builder, featureGroupCount): |
| return StablehloConvolutionOptionsAddFeatureGroupCount(builder, featureGroupCount) |
| def StablehloConvolutionOptionsAddBatchGroupCount(builder, batchGroupCount): builder.PrependInt64Slot(15, batchGroupCount, 0) |
| def AddBatchGroupCount(builder, batchGroupCount): |
| return StablehloConvolutionOptionsAddBatchGroupCount(builder, batchGroupCount) |
| def StablehloConvolutionOptionsAddPrecisionConfig(builder, precisionConfig): builder.PrependUOffsetTRelativeSlot(16, flatbuffers.number_types.UOffsetTFlags.py_type(precisionConfig), 0) |
| def AddPrecisionConfig(builder, precisionConfig): |
| return StablehloConvolutionOptionsAddPrecisionConfig(builder, precisionConfig) |
| def StablehloConvolutionOptionsStartPrecisionConfigVector(builder, numElems): return builder.StartVector(4, numElems, 4) |
| def StartPrecisionConfigVector(builder, numElems): |
| return StablehloConvolutionOptionsStartPrecisionConfigVector(builder, numElems) |
| def StablehloConvolutionOptionsEnd(builder): return builder.EndObject() |
| def End(builder): |
| return StablehloConvolutionOptionsEnd(builder) |