18164 lines
615 KiB
Python
18164 lines
615 KiB
Python
import flatbuffers
|
|
|
|
# automatically generated by the FlatBuffers compiler, do not modify
|
|
|
|
# namespace: tflite
|
|
|
|
from flatbuffers.compat import import_numpy
|
|
np = import_numpy()
|
|
|
|
class TensorType(object):
|
|
FLOAT32 = 0
|
|
FLOAT16 = 1
|
|
INT32 = 2
|
|
UINT8 = 3
|
|
INT64 = 4
|
|
STRING = 5
|
|
BOOL = 6
|
|
INT16 = 7
|
|
COMPLEX64 = 8
|
|
INT8 = 9
|
|
FLOAT64 = 10
|
|
COMPLEX128 = 11
|
|
UINT64 = 12
|
|
RESOURCE = 13
|
|
VARIANT = 14
|
|
UINT32 = 15
|
|
UINT16 = 16
|
|
INT4 = 17
|
|
|
|
|
|
class QuantizationDetails(object):
|
|
NONE = 0
|
|
CustomQuantization = 1
|
|
|
|
def QuantizationDetailsCreator(unionType, table):
|
|
from flatbuffers.table import Table
|
|
if not isinstance(table, Table):
|
|
return None
|
|
if unionType == QuantizationDetails().CustomQuantization:
|
|
return CustomQuantizationT.InitFromBuf(table.Bytes, table.Pos)
|
|
return None
|
|
|
|
|
|
class DimensionType(object):
|
|
DENSE = 0
|
|
SPARSE_CSR = 1
|
|
|
|
|
|
class SparseIndexVector(object):
|
|
NONE = 0
|
|
Int32Vector = 1
|
|
Uint16Vector = 2
|
|
Uint8Vector = 3
|
|
|
|
def SparseIndexVectorCreator(unionType, table):
|
|
from flatbuffers.table import Table
|
|
if not isinstance(table, Table):
|
|
return None
|
|
if unionType == SparseIndexVector().Int32Vector:
|
|
return Int32VectorT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == SparseIndexVector().Uint16Vector:
|
|
return Uint16VectorT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == SparseIndexVector().Uint8Vector:
|
|
return Uint8VectorT.InitFromBuf(table.Bytes, table.Pos)
|
|
return None
|
|
|
|
|
|
class BuiltinOperator(object):
|
|
ADD = 0
|
|
AVERAGE_POOL_2D = 1
|
|
CONCATENATION = 2
|
|
CONV_2D = 3
|
|
DEPTHWISE_CONV_2D = 4
|
|
DEPTH_TO_SPACE = 5
|
|
DEQUANTIZE = 6
|
|
EMBEDDING_LOOKUP = 7
|
|
FLOOR = 8
|
|
FULLY_CONNECTED = 9
|
|
HASHTABLE_LOOKUP = 10
|
|
L2_NORMALIZATION = 11
|
|
L2_POOL_2D = 12
|
|
LOCAL_RESPONSE_NORMALIZATION = 13
|
|
LOGISTIC = 14
|
|
LSH_PROJECTION = 15
|
|
LSTM = 16
|
|
MAX_POOL_2D = 17
|
|
MUL = 18
|
|
RELU = 19
|
|
RELU_N1_TO_1 = 20
|
|
RELU6 = 21
|
|
RESHAPE = 22
|
|
RESIZE_BILINEAR = 23
|
|
RNN = 24
|
|
SOFTMAX = 25
|
|
SPACE_TO_DEPTH = 26
|
|
SVDF = 27
|
|
TANH = 28
|
|
CONCAT_EMBEDDINGS = 29
|
|
SKIP_GRAM = 30
|
|
CALL = 31
|
|
CUSTOM = 32
|
|
EMBEDDING_LOOKUP_SPARSE = 33
|
|
PAD = 34
|
|
UNIDIRECTIONAL_SEQUENCE_RNN = 35
|
|
GATHER = 36
|
|
BATCH_TO_SPACE_ND = 37
|
|
SPACE_TO_BATCH_ND = 38
|
|
TRANSPOSE = 39
|
|
MEAN = 40
|
|
SUB = 41
|
|
DIV = 42
|
|
SQUEEZE = 43
|
|
UNIDIRECTIONAL_SEQUENCE_LSTM = 44
|
|
STRIDED_SLICE = 45
|
|
BIDIRECTIONAL_SEQUENCE_RNN = 46
|
|
EXP = 47
|
|
TOPK_V2 = 48
|
|
SPLIT = 49
|
|
LOG_SOFTMAX = 50
|
|
DELEGATE = 51
|
|
BIDIRECTIONAL_SEQUENCE_LSTM = 52
|
|
CAST = 53
|
|
PRELU = 54
|
|
MAXIMUM = 55
|
|
ARG_MAX = 56
|
|
MINIMUM = 57
|
|
LESS = 58
|
|
NEG = 59
|
|
PADV2 = 60
|
|
GREATER = 61
|
|
GREATER_EQUAL = 62
|
|
LESS_EQUAL = 63
|
|
SELECT = 64
|
|
SLICE = 65
|
|
SIN = 66
|
|
TRANSPOSE_CONV = 67
|
|
SPARSE_TO_DENSE = 68
|
|
TILE = 69
|
|
EXPAND_DIMS = 70
|
|
EQUAL = 71
|
|
NOT_EQUAL = 72
|
|
LOG = 73
|
|
SUM = 74
|
|
SQRT = 75
|
|
RSQRT = 76
|
|
SHAPE = 77
|
|
POW = 78
|
|
ARG_MIN = 79
|
|
FAKE_QUANT = 80
|
|
REDUCE_PROD = 81
|
|
REDUCE_MAX = 82
|
|
PACK = 83
|
|
LOGICAL_OR = 84
|
|
ONE_HOT = 85
|
|
LOGICAL_AND = 86
|
|
LOGICAL_NOT = 87
|
|
UNPACK = 88
|
|
REDUCE_MIN = 89
|
|
FLOOR_DIV = 90
|
|
REDUCE_ANY = 91
|
|
SQUARE = 92
|
|
ZEROS_LIKE = 93
|
|
FILL = 94
|
|
FLOOR_MOD = 95
|
|
RANGE = 96
|
|
RESIZE_NEAREST_NEIGHBOR = 97
|
|
LEAKY_RELU = 98
|
|
SQUARED_DIFFERENCE = 99
|
|
MIRROR_PAD = 100
|
|
ABS = 101
|
|
SPLIT_V = 102
|
|
UNIQUE = 103
|
|
CEIL = 104
|
|
REVERSE_V2 = 105
|
|
ADD_N = 106
|
|
GATHER_ND = 107
|
|
COS = 108
|
|
WHERE = 109
|
|
RANK = 110
|
|
ELU = 111
|
|
REVERSE_SEQUENCE = 112
|
|
MATRIX_DIAG = 113
|
|
QUANTIZE = 114
|
|
MATRIX_SET_DIAG = 115
|
|
ROUND = 116
|
|
HARD_SWISH = 117
|
|
IF = 118
|
|
WHILE = 119
|
|
NON_MAX_SUPPRESSION_V4 = 120
|
|
NON_MAX_SUPPRESSION_V5 = 121
|
|
SCATTER_ND = 122
|
|
SELECT_V2 = 123
|
|
DENSIFY = 124
|
|
SEGMENT_SUM = 125
|
|
BATCH_MATMUL = 126
|
|
PLACEHOLDER_FOR_GREATER_OP_CODES = 127
|
|
CUMSUM = 128
|
|
CALL_ONCE = 129
|
|
BROADCAST_TO = 130
|
|
RFFT2D = 131
|
|
CONV_3D = 132
|
|
IMAG = 133
|
|
REAL = 134
|
|
COMPLEX_ABS = 135
|
|
HASHTABLE = 136
|
|
HASHTABLE_FIND = 137
|
|
HASHTABLE_IMPORT = 138
|
|
HASHTABLE_SIZE = 139
|
|
REDUCE_ALL = 140
|
|
CONV_3D_TRANSPOSE = 141
|
|
VAR_HANDLE = 142
|
|
READ_VARIABLE = 143
|
|
ASSIGN_VARIABLE = 144
|
|
BROADCAST_ARGS = 145
|
|
RANDOM_STANDARD_NORMAL = 146
|
|
BUCKETIZE = 147
|
|
RANDOM_UNIFORM = 148
|
|
MULTINOMIAL = 149
|
|
GELU = 150
|
|
DYNAMIC_UPDATE_SLICE = 151
|
|
RELU_0_TO_1 = 152
|
|
UNSORTED_SEGMENT_PROD = 153
|
|
UNSORTED_SEGMENT_MAX = 154
|
|
UNSORTED_SEGMENT_SUM = 155
|
|
ATAN2 = 156
|
|
UNSORTED_SEGMENT_MIN = 157
|
|
SIGN = 158
|
|
BITCAST = 159
|
|
BITWISE_XOR = 160
|
|
RIGHT_SHIFT = 161
|
|
STABLEHLO_LOGISTIC = 162
|
|
STABLEHLO_ADD = 163
|
|
STABLEHLO_DIVIDE = 164
|
|
STABLEHLO_MULTIPLY = 165
|
|
STABLEHLO_MAXIMUM = 166
|
|
STABLEHLO_RESHAPE = 167
|
|
STABLEHLO_CLAMP = 168
|
|
STABLEHLO_CONCATENATE = 169
|
|
STABLEHLO_BROADCAST_IN_DIM = 170
|
|
STABLEHLO_CONVOLUTION = 171
|
|
STABLEHLO_SLICE = 172
|
|
STABLEHLO_CUSTOM_CALL = 173
|
|
STABLEHLO_REDUCE = 174
|
|
STABLEHLO_ABS = 175
|
|
STABLEHLO_AND = 176
|
|
STABLEHLO_COSINE = 177
|
|
STABLEHLO_EXPONENTIAL = 178
|
|
STABLEHLO_FLOOR = 179
|
|
STABLEHLO_LOG = 180
|
|
STABLEHLO_MINIMUM = 181
|
|
STABLEHLO_NEGATE = 182
|
|
STABLEHLO_OR = 183
|
|
STABLEHLO_POWER = 184
|
|
STABLEHLO_REMAINDER = 185
|
|
STABLEHLO_RSQRT = 186
|
|
STABLEHLO_SELECT = 187
|
|
STABLEHLO_SUBTRACT = 188
|
|
STABLEHLO_TANH = 189
|
|
STABLEHLO_SCATTER = 190
|
|
STABLEHLO_COMPARE = 191
|
|
STABLEHLO_CONVERT = 192
|
|
STABLEHLO_DYNAMIC_SLICE = 193
|
|
STABLEHLO_DYNAMIC_UPDATE_SLICE = 194
|
|
STABLEHLO_PAD = 195
|
|
STABLEHLO_IOTA = 196
|
|
STABLEHLO_DOT_GENERAL = 197
|
|
STABLEHLO_REDUCE_WINDOW = 198
|
|
STABLEHLO_SORT = 199
|
|
STABLEHLO_WHILE = 200
|
|
STABLEHLO_GATHER = 201
|
|
STABLEHLO_TRANSPOSE = 202
|
|
DILATE = 203
|
|
STABLEHLO_RNG_BIT_GENERATOR = 204
|
|
REDUCE_WINDOW = 205
|
|
|
|
|
|
class BuiltinOptions(object):
|
|
NONE = 0
|
|
Conv2DOptions = 1
|
|
DepthwiseConv2DOptions = 2
|
|
ConcatEmbeddingsOptions = 3
|
|
LSHProjectionOptions = 4
|
|
Pool2DOptions = 5
|
|
SVDFOptions = 6
|
|
RNNOptions = 7
|
|
FullyConnectedOptions = 8
|
|
SoftmaxOptions = 9
|
|
ConcatenationOptions = 10
|
|
AddOptions = 11
|
|
L2NormOptions = 12
|
|
LocalResponseNormalizationOptions = 13
|
|
LSTMOptions = 14
|
|
ResizeBilinearOptions = 15
|
|
CallOptions = 16
|
|
ReshapeOptions = 17
|
|
SkipGramOptions = 18
|
|
SpaceToDepthOptions = 19
|
|
EmbeddingLookupSparseOptions = 20
|
|
MulOptions = 21
|
|
PadOptions = 22
|
|
GatherOptions = 23
|
|
BatchToSpaceNDOptions = 24
|
|
SpaceToBatchNDOptions = 25
|
|
TransposeOptions = 26
|
|
ReducerOptions = 27
|
|
SubOptions = 28
|
|
DivOptions = 29
|
|
SqueezeOptions = 30
|
|
SequenceRNNOptions = 31
|
|
StridedSliceOptions = 32
|
|
ExpOptions = 33
|
|
TopKV2Options = 34
|
|
SplitOptions = 35
|
|
LogSoftmaxOptions = 36
|
|
CastOptions = 37
|
|
DequantizeOptions = 38
|
|
MaximumMinimumOptions = 39
|
|
ArgMaxOptions = 40
|
|
LessOptions = 41
|
|
NegOptions = 42
|
|
PadV2Options = 43
|
|
GreaterOptions = 44
|
|
GreaterEqualOptions = 45
|
|
LessEqualOptions = 46
|
|
SelectOptions = 47
|
|
SliceOptions = 48
|
|
TransposeConvOptions = 49
|
|
SparseToDenseOptions = 50
|
|
TileOptions = 51
|
|
ExpandDimsOptions = 52
|
|
EqualOptions = 53
|
|
NotEqualOptions = 54
|
|
ShapeOptions = 55
|
|
PowOptions = 56
|
|
ArgMinOptions = 57
|
|
FakeQuantOptions = 58
|
|
PackOptions = 59
|
|
LogicalOrOptions = 60
|
|
OneHotOptions = 61
|
|
LogicalAndOptions = 62
|
|
LogicalNotOptions = 63
|
|
UnpackOptions = 64
|
|
FloorDivOptions = 65
|
|
SquareOptions = 66
|
|
ZerosLikeOptions = 67
|
|
FillOptions = 68
|
|
BidirectionalSequenceLSTMOptions = 69
|
|
BidirectionalSequenceRNNOptions = 70
|
|
UnidirectionalSequenceLSTMOptions = 71
|
|
FloorModOptions = 72
|
|
RangeOptions = 73
|
|
ResizeNearestNeighborOptions = 74
|
|
LeakyReluOptions = 75
|
|
SquaredDifferenceOptions = 76
|
|
MirrorPadOptions = 77
|
|
AbsOptions = 78
|
|
SplitVOptions = 79
|
|
UniqueOptions = 80
|
|
ReverseV2Options = 81
|
|
AddNOptions = 82
|
|
GatherNdOptions = 83
|
|
CosOptions = 84
|
|
WhereOptions = 85
|
|
RankOptions = 86
|
|
ReverseSequenceOptions = 87
|
|
MatrixDiagOptions = 88
|
|
QuantizeOptions = 89
|
|
MatrixSetDiagOptions = 90
|
|
HardSwishOptions = 91
|
|
IfOptions = 92
|
|
WhileOptions = 93
|
|
DepthToSpaceOptions = 94
|
|
NonMaxSuppressionV4Options = 95
|
|
NonMaxSuppressionV5Options = 96
|
|
ScatterNdOptions = 97
|
|
SelectV2Options = 98
|
|
DensifyOptions = 99
|
|
SegmentSumOptions = 100
|
|
BatchMatMulOptions = 101
|
|
CumsumOptions = 102
|
|
CallOnceOptions = 103
|
|
BroadcastToOptions = 104
|
|
Rfft2dOptions = 105
|
|
Conv3DOptions = 106
|
|
HashtableOptions = 107
|
|
HashtableFindOptions = 108
|
|
HashtableImportOptions = 109
|
|
HashtableSizeOptions = 110
|
|
VarHandleOptions = 111
|
|
ReadVariableOptions = 112
|
|
AssignVariableOptions = 113
|
|
RandomOptions = 114
|
|
BucketizeOptions = 115
|
|
GeluOptions = 116
|
|
DynamicUpdateSliceOptions = 117
|
|
UnsortedSegmentProdOptions = 118
|
|
UnsortedSegmentMaxOptions = 119
|
|
UnsortedSegmentMinOptions = 120
|
|
UnsortedSegmentSumOptions = 121
|
|
ATan2Options = 122
|
|
SignOptions = 123
|
|
BitcastOptions = 124
|
|
BitwiseXorOptions = 125
|
|
RightShiftOptions = 126
|
|
|
|
def BuiltinOptionsCreator(unionType, table):
|
|
from flatbuffers.table import Table
|
|
if not isinstance(table, Table):
|
|
return None
|
|
if unionType == BuiltinOptions().Conv2DOptions:
|
|
return Conv2DOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DepthwiseConv2DOptions:
|
|
return DepthwiseConv2DOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ConcatEmbeddingsOptions:
|
|
return ConcatEmbeddingsOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LSHProjectionOptions:
|
|
return LSHProjectionOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().Pool2DOptions:
|
|
return Pool2DOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SVDFOptions:
|
|
return SVDFOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().RNNOptions:
|
|
return RNNOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().FullyConnectedOptions:
|
|
return FullyConnectedOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SoftmaxOptions:
|
|
return SoftmaxOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ConcatenationOptions:
|
|
return ConcatenationOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().AddOptions:
|
|
return AddOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().L2NormOptions:
|
|
return L2NormOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LocalResponseNormalizationOptions:
|
|
return LocalResponseNormalizationOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LSTMOptions:
|
|
return LSTMOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ResizeBilinearOptions:
|
|
return ResizeBilinearOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().CallOptions:
|
|
return CallOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ReshapeOptions:
|
|
return ReshapeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SkipGramOptions:
|
|
return SkipGramOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SpaceToDepthOptions:
|
|
return SpaceToDepthOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().EmbeddingLookupSparseOptions:
|
|
return EmbeddingLookupSparseOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().MulOptions:
|
|
return MulOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().PadOptions:
|
|
return PadOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().GatherOptions:
|
|
return GatherOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BatchToSpaceNDOptions:
|
|
return BatchToSpaceNDOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SpaceToBatchNDOptions:
|
|
return SpaceToBatchNDOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().TransposeOptions:
|
|
return TransposeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ReducerOptions:
|
|
return ReducerOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SubOptions:
|
|
return SubOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DivOptions:
|
|
return DivOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SqueezeOptions:
|
|
return SqueezeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SequenceRNNOptions:
|
|
return SequenceRNNOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().StridedSliceOptions:
|
|
return StridedSliceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ExpOptions:
|
|
return ExpOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().TopKV2Options:
|
|
return TopKV2OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SplitOptions:
|
|
return SplitOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LogSoftmaxOptions:
|
|
return LogSoftmaxOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().CastOptions:
|
|
return CastOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DequantizeOptions:
|
|
return DequantizeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().MaximumMinimumOptions:
|
|
return MaximumMinimumOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ArgMaxOptions:
|
|
return ArgMaxOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LessOptions:
|
|
return LessOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().NegOptions:
|
|
return NegOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().PadV2Options:
|
|
return PadV2OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().GreaterOptions:
|
|
return GreaterOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().GreaterEqualOptions:
|
|
return GreaterEqualOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LessEqualOptions:
|
|
return LessEqualOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SelectOptions:
|
|
return SelectOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SliceOptions:
|
|
return SliceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().TransposeConvOptions:
|
|
return TransposeConvOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SparseToDenseOptions:
|
|
return SparseToDenseOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().TileOptions:
|
|
return TileOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ExpandDimsOptions:
|
|
return ExpandDimsOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().EqualOptions:
|
|
return EqualOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().NotEqualOptions:
|
|
return NotEqualOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ShapeOptions:
|
|
return ShapeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().PowOptions:
|
|
return PowOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ArgMinOptions:
|
|
return ArgMinOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().FakeQuantOptions:
|
|
return FakeQuantOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().PackOptions:
|
|
return PackOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LogicalOrOptions:
|
|
return LogicalOrOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().OneHotOptions:
|
|
return OneHotOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LogicalAndOptions:
|
|
return LogicalAndOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LogicalNotOptions:
|
|
return LogicalNotOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnpackOptions:
|
|
return UnpackOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().FloorDivOptions:
|
|
return FloorDivOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SquareOptions:
|
|
return SquareOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ZerosLikeOptions:
|
|
return ZerosLikeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().FillOptions:
|
|
return FillOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BidirectionalSequenceLSTMOptions:
|
|
return BidirectionalSequenceLSTMOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BidirectionalSequenceRNNOptions:
|
|
return BidirectionalSequenceRNNOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnidirectionalSequenceLSTMOptions:
|
|
return UnidirectionalSequenceLSTMOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().FloorModOptions:
|
|
return FloorModOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().RangeOptions:
|
|
return RangeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ResizeNearestNeighborOptions:
|
|
return ResizeNearestNeighborOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().LeakyReluOptions:
|
|
return LeakyReluOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SquaredDifferenceOptions:
|
|
return SquaredDifferenceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().MirrorPadOptions:
|
|
return MirrorPadOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().AbsOptions:
|
|
return AbsOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SplitVOptions:
|
|
return SplitVOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UniqueOptions:
|
|
return UniqueOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ReverseV2Options:
|
|
return ReverseV2OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().AddNOptions:
|
|
return AddNOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().GatherNdOptions:
|
|
return GatherNdOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().CosOptions:
|
|
return CosOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().WhereOptions:
|
|
return WhereOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().RankOptions:
|
|
return RankOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ReverseSequenceOptions:
|
|
return ReverseSequenceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().MatrixDiagOptions:
|
|
return MatrixDiagOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().QuantizeOptions:
|
|
return QuantizeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().MatrixSetDiagOptions:
|
|
return MatrixSetDiagOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().HardSwishOptions:
|
|
return HardSwishOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().IfOptions:
|
|
return IfOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().WhileOptions:
|
|
return WhileOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DepthToSpaceOptions:
|
|
return DepthToSpaceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().NonMaxSuppressionV4Options:
|
|
return NonMaxSuppressionV4OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().NonMaxSuppressionV5Options:
|
|
return NonMaxSuppressionV5OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ScatterNdOptions:
|
|
return ScatterNdOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SelectV2Options:
|
|
return SelectV2OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DensifyOptions:
|
|
return DensifyOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SegmentSumOptions:
|
|
return SegmentSumOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BatchMatMulOptions:
|
|
return BatchMatMulOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().CumsumOptions:
|
|
return CumsumOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().CallOnceOptions:
|
|
return CallOnceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BroadcastToOptions:
|
|
return BroadcastToOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().Rfft2dOptions:
|
|
return Rfft2dOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().Conv3DOptions:
|
|
return Conv3DOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().HashtableOptions:
|
|
return HashtableOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().HashtableFindOptions:
|
|
return HashtableFindOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().HashtableImportOptions:
|
|
return HashtableImportOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().HashtableSizeOptions:
|
|
return HashtableSizeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().VarHandleOptions:
|
|
return VarHandleOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ReadVariableOptions:
|
|
return ReadVariableOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().AssignVariableOptions:
|
|
return AssignVariableOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().RandomOptions:
|
|
return RandomOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BucketizeOptions:
|
|
return BucketizeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().GeluOptions:
|
|
return GeluOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().DynamicUpdateSliceOptions:
|
|
return DynamicUpdateSliceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnsortedSegmentProdOptions:
|
|
return UnsortedSegmentProdOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnsortedSegmentMaxOptions:
|
|
return UnsortedSegmentMaxOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnsortedSegmentMinOptions:
|
|
return UnsortedSegmentMinOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().UnsortedSegmentSumOptions:
|
|
return UnsortedSegmentSumOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().ATan2Options:
|
|
return ATan2OptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().SignOptions:
|
|
return SignOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BitcastOptions:
|
|
return BitcastOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().BitwiseXorOptions:
|
|
return BitwiseXorOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions().RightShiftOptions:
|
|
return RightShiftOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
return None
|
|
|
|
|
|
class BuiltinOptions2(object):
|
|
NONE = 0
|
|
StablehloConcatenateOptions = 1
|
|
StablehloBroadcastInDimOptions = 2
|
|
StablehloSliceOptions = 3
|
|
StablehloConvolutionOptions = 4
|
|
StablehloCustomCallOptions = 5
|
|
StablehloReduceOptions = 6
|
|
StablehloScatterOptions = 7
|
|
StablehloCompareOptions = 8
|
|
StablehloDynamicSliceOptions = 9
|
|
StablehloPadOptions = 10
|
|
StablehloIotaOptions = 11
|
|
StablehloDotGeneralOptions = 12
|
|
StablehloReduceWindowOptions = 13
|
|
StablehloSortOptions = 14
|
|
StablehloWhileOptions = 15
|
|
StablehloGatherOptions = 16
|
|
StablehloTransposeOptions = 17
|
|
DilateOptions = 18
|
|
StablehloRngBitGeneratorOptions = 19
|
|
ReduceWindowOptions = 20
|
|
|
|
def BuiltinOptions2Creator(unionType, table):
|
|
from flatbuffers.table import Table
|
|
if not isinstance(table, Table):
|
|
return None
|
|
if unionType == BuiltinOptions2().StablehloConcatenateOptions:
|
|
return StablehloConcatenateOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloBroadcastInDimOptions:
|
|
return StablehloBroadcastInDimOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloSliceOptions:
|
|
return StablehloSliceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloConvolutionOptions:
|
|
return StablehloConvolutionOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloCustomCallOptions:
|
|
return StablehloCustomCallOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloReduceOptions:
|
|
return StablehloReduceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloScatterOptions:
|
|
return StablehloScatterOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloCompareOptions:
|
|
return StablehloCompareOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloDynamicSliceOptions:
|
|
return StablehloDynamicSliceOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloPadOptions:
|
|
return StablehloPadOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloIotaOptions:
|
|
return StablehloIotaOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloDotGeneralOptions:
|
|
return StablehloDotGeneralOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloReduceWindowOptions:
|
|
return StablehloReduceWindowOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloSortOptions:
|
|
return StablehloSortOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloWhileOptions:
|
|
return StablehloWhileOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloGatherOptions:
|
|
return StablehloGatherOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloTransposeOptions:
|
|
return StablehloTransposeOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().DilateOptions:
|
|
return DilateOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().StablehloRngBitGeneratorOptions:
|
|
return StablehloRngBitGeneratorOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
if unionType == BuiltinOptions2().ReduceWindowOptions:
|
|
return ReduceWindowOptionsT.InitFromBuf(table.Bytes, table.Pos)
|
|
return None
|
|
|
|
|
|
class StablehloPrecisionConfig(object):
|
|
DEFAULT = 0
|
|
HIGH = 1
|
|
HIGHEST = 2
|
|
|
|
|
|
class StablehloComparisonDirection(object):
|
|
STABLEHLO_COMPARISON_DIRECTION_EQ = 0
|
|
STABLEHLO_COMPARISON_DIRECTION_NE = 1
|
|
STABLEHLO_COMPARISON_DIRECTION_GE = 2
|
|
STABLEHLO_COMPARISON_DIRECTION_GT = 3
|
|
STABLEHLO_COMPARISON_DIRECTION_LE = 4
|
|
STABLEHLO_COMPARISON_DIRECTION_LT = 5
|
|
|
|
|
|
class StablehloComparisonType(object):
|
|
STABLEHLO_COMPARISON_TYPE_NOTYPE = 0
|
|
STABLEHLO_COMPARISON_TYPE_FLOAT = 1
|
|
STABLEHLO_COMPARISON_TYPE_FLOAT_TOTAL_ORDER = 2
|
|
STABLEHLO_COMPARISON_TYPE_SIGNED = 3
|
|
STABLEHLO_COMPARISON_TYPE_UNSIGNED = 4
|
|
|
|
|
|
class RngAlgorithm(object):
|
|
DEFAULT = 0
|
|
PHILOX = 1
|
|
THREEFRY = 2
|
|
|
|
|
|
class Padding(object):
|
|
SAME = 0
|
|
VALID = 1
|
|
|
|
|
|
class ActivationFunctionType(object):
|
|
NONE = 0
|
|
RELU = 1
|
|
RELU_N1_TO_1 = 2
|
|
RELU6 = 3
|
|
TANH = 4
|
|
SIGN_BIT = 5
|
|
|
|
|
|
class LSHProjectionType(object):
|
|
UNKNOWN = 0
|
|
SPARSE = 1
|
|
DENSE = 2
|
|
|
|
|
|
class FullyConnectedOptionsWeightsFormat(object):
|
|
DEFAULT = 0
|
|
SHUFFLED4x16INT8 = 1
|
|
|
|
|
|
class LSTMKernelType(object):
|
|
FULL = 0
|
|
BASIC = 1
|
|
|
|
|
|
class CombinerType(object):
|
|
SUM = 0
|
|
MEAN = 1
|
|
SQRTN = 2
|
|
|
|
|
|
class MirrorPadMode(object):
|
|
REFLECT = 0
|
|
SYMMETRIC = 1
|
|
|
|
|
|
class ReduceWindowFunction(object):
|
|
UNSUPPORTED = 0
|
|
ADD = 1
|
|
MUL = 2
|
|
MINIMUM = 3
|
|
MAXIMUM = 4
|
|
ALL = 5
|
|
ANY = 6
|
|
|
|
|
|
class CustomOptionsFormat(object):
|
|
FLEXBUFFERS = 0
|
|
|
|
|
|
class CustomQuantization(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CustomQuantization()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCustomQuantization(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CustomQuantizationBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CustomQuantization
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# CustomQuantization
|
|
def Custom(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.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# CustomQuantization
|
|
def CustomAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o)
|
|
return 0
|
|
|
|
# CustomQuantization
|
|
def CustomLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# CustomQuantization
|
|
def CustomIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def CustomQuantizationStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def CustomQuantizationAddCustom(builder, custom):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(custom), 0)
|
|
|
|
def CustomQuantizationStartCustomVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def CustomQuantizationEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class CustomQuantizationT(object):
|
|
|
|
# CustomQuantizationT
|
|
def __init__(self):
|
|
self.custom = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
customQuantization = CustomQuantization()
|
|
customQuantization.Init(buf, pos)
|
|
return cls.InitFromObj(customQuantization)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, customQuantization):
|
|
x = CustomQuantizationT()
|
|
x._UnPack(customQuantization)
|
|
return x
|
|
|
|
# CustomQuantizationT
|
|
def _UnPack(self, customQuantization):
|
|
if customQuantization is None:
|
|
return
|
|
if not customQuantization.CustomIsNone():
|
|
if np is None:
|
|
self.custom = []
|
|
for i in range(customQuantization.CustomLength()):
|
|
self.custom.append(customQuantization.Custom(i))
|
|
else:
|
|
self.custom = customQuantization.CustomAsNumpy()
|
|
|
|
# CustomQuantizationT
|
|
def Pack(self, builder):
|
|
if self.custom is not None:
|
|
if np is not None and type(self.custom) is np.ndarray:
|
|
custom = builder.CreateNumpyVector(self.custom)
|
|
else:
|
|
CustomQuantizationStartCustomVector(builder, len(self.custom))
|
|
for i in reversed(range(len(self.custom))):
|
|
builder.PrependUint8(self.custom[i])
|
|
custom = builder.EndVector()
|
|
CustomQuantizationStart(builder)
|
|
if self.custom is not None:
|
|
CustomQuantizationAddCustom(builder, custom)
|
|
customQuantization = CustomQuantizationEnd(builder)
|
|
return customQuantization
|
|
|
|
|
|
class QuantizationParameters(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = QuantizationParameters()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsQuantizationParameters(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def QuantizationParametersBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# QuantizationParameters
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# QuantizationParameters
|
|
def Min(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.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MinAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MinLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MinIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# QuantizationParameters
|
|
def Max(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.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MaxAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MaxLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def MaxIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# QuantizationParameters
|
|
def Scale(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.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def ScaleAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def ScaleLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def ScaleIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# QuantizationParameters
|
|
def ZeroPoint(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
|
|
|
|
# QuantizationParameters
|
|
def ZeroPointAsNumpy(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
|
|
|
|
# QuantizationParameters
|
|
def ZeroPointLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def ZeroPointIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
return o == 0
|
|
|
|
# QuantizationParameters
|
|
def DetailsType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# QuantizationParameters
|
|
def Details(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
from flatbuffers.table import Table
|
|
obj = Table(bytearray(), 0)
|
|
self._tab.Union(obj, o)
|
|
return obj
|
|
return None
|
|
|
|
# QuantizationParameters
|
|
def QuantizedDimension(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def QuantizationParametersStart(builder):
|
|
builder.StartObject(7)
|
|
|
|
def QuantizationParametersAddMin(builder, min):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(min), 0)
|
|
|
|
def QuantizationParametersStartMinVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def QuantizationParametersAddMax(builder, max):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(max), 0)
|
|
|
|
def QuantizationParametersStartMaxVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def QuantizationParametersAddScale(builder, scale):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(scale), 0)
|
|
|
|
def QuantizationParametersStartScaleVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def QuantizationParametersAddZeroPoint(builder, zeroPoint):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(zeroPoint), 0)
|
|
|
|
def QuantizationParametersStartZeroPointVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def QuantizationParametersAddDetailsType(builder, detailsType):
|
|
builder.PrependUint8Slot(4, detailsType, 0)
|
|
|
|
def QuantizationParametersAddDetails(builder, details):
|
|
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(details), 0)
|
|
|
|
def QuantizationParametersAddQuantizedDimension(builder, quantizedDimension):
|
|
builder.PrependInt32Slot(6, quantizedDimension, 0)
|
|
|
|
def QuantizationParametersEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List, Union
|
|
except:
|
|
pass
|
|
|
|
class QuantizationParametersT(object):
|
|
|
|
# QuantizationParametersT
|
|
def __init__(self):
|
|
self.min = None # type: List[float]
|
|
self.max = None # type: List[float]
|
|
self.scale = None # type: List[float]
|
|
self.zeroPoint = None # type: List[int]
|
|
self.detailsType = 0 # type: int
|
|
self.details = None # type: Union[None, CustomQuantizationT]
|
|
self.quantizedDimension = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
quantizationParameters = QuantizationParameters()
|
|
quantizationParameters.Init(buf, pos)
|
|
return cls.InitFromObj(quantizationParameters)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, quantizationParameters):
|
|
x = QuantizationParametersT()
|
|
x._UnPack(quantizationParameters)
|
|
return x
|
|
|
|
# QuantizationParametersT
|
|
def _UnPack(self, quantizationParameters):
|
|
if quantizationParameters is None:
|
|
return
|
|
if not quantizationParameters.MinIsNone():
|
|
if np is None:
|
|
self.min = []
|
|
for i in range(quantizationParameters.MinLength()):
|
|
self.min.append(quantizationParameters.Min(i))
|
|
else:
|
|
self.min = quantizationParameters.MinAsNumpy()
|
|
if not quantizationParameters.MaxIsNone():
|
|
if np is None:
|
|
self.max = []
|
|
for i in range(quantizationParameters.MaxLength()):
|
|
self.max.append(quantizationParameters.Max(i))
|
|
else:
|
|
self.max = quantizationParameters.MaxAsNumpy()
|
|
if not quantizationParameters.ScaleIsNone():
|
|
if np is None:
|
|
self.scale = []
|
|
for i in range(quantizationParameters.ScaleLength()):
|
|
self.scale.append(quantizationParameters.Scale(i))
|
|
else:
|
|
self.scale = quantizationParameters.ScaleAsNumpy()
|
|
if not quantizationParameters.ZeroPointIsNone():
|
|
if np is None:
|
|
self.zeroPoint = []
|
|
for i in range(quantizationParameters.ZeroPointLength()):
|
|
self.zeroPoint.append(quantizationParameters.ZeroPoint(i))
|
|
else:
|
|
self.zeroPoint = quantizationParameters.ZeroPointAsNumpy()
|
|
self.detailsType = quantizationParameters.DetailsType()
|
|
self.details = QuantizationDetailsCreator(self.detailsType, quantizationParameters.Details())
|
|
self.quantizedDimension = quantizationParameters.QuantizedDimension()
|
|
|
|
# QuantizationParametersT
|
|
def Pack(self, builder):
|
|
if self.min is not None:
|
|
if np is not None and type(self.min) is np.ndarray:
|
|
min = builder.CreateNumpyVector(self.min)
|
|
else:
|
|
QuantizationParametersStartMinVector(builder, len(self.min))
|
|
for i in reversed(range(len(self.min))):
|
|
builder.PrependFloat32(self.min[i])
|
|
min = builder.EndVector()
|
|
if self.max is not None:
|
|
if np is not None and type(self.max) is np.ndarray:
|
|
max = builder.CreateNumpyVector(self.max)
|
|
else:
|
|
QuantizationParametersStartMaxVector(builder, len(self.max))
|
|
for i in reversed(range(len(self.max))):
|
|
builder.PrependFloat32(self.max[i])
|
|
max = builder.EndVector()
|
|
if self.scale is not None:
|
|
if np is not None and type(self.scale) is np.ndarray:
|
|
scale = builder.CreateNumpyVector(self.scale)
|
|
else:
|
|
QuantizationParametersStartScaleVector(builder, len(self.scale))
|
|
for i in reversed(range(len(self.scale))):
|
|
builder.PrependFloat32(self.scale[i])
|
|
scale = builder.EndVector()
|
|
if self.zeroPoint is not None:
|
|
if np is not None and type(self.zeroPoint) is np.ndarray:
|
|
zeroPoint = builder.CreateNumpyVector(self.zeroPoint)
|
|
else:
|
|
QuantizationParametersStartZeroPointVector(builder, len(self.zeroPoint))
|
|
for i in reversed(range(len(self.zeroPoint))):
|
|
builder.PrependInt64(self.zeroPoint[i])
|
|
zeroPoint = builder.EndVector()
|
|
if self.details is not None:
|
|
details = self.details.Pack(builder)
|
|
QuantizationParametersStart(builder)
|
|
if self.min is not None:
|
|
QuantizationParametersAddMin(builder, min)
|
|
if self.max is not None:
|
|
QuantizationParametersAddMax(builder, max)
|
|
if self.scale is not None:
|
|
QuantizationParametersAddScale(builder, scale)
|
|
if self.zeroPoint is not None:
|
|
QuantizationParametersAddZeroPoint(builder, zeroPoint)
|
|
QuantizationParametersAddDetailsType(builder, self.detailsType)
|
|
if self.details is not None:
|
|
QuantizationParametersAddDetails(builder, details)
|
|
QuantizationParametersAddQuantizedDimension(builder, self.quantizedDimension)
|
|
quantizationParameters = QuantizationParametersEnd(builder)
|
|
return quantizationParameters
|
|
|
|
|
|
class Int32Vector(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Int32Vector()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsInt32Vector(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Int32VectorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Int32Vector
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Int32Vector
|
|
def Values(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Int32Vector
|
|
def ValuesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Int32Vector
|
|
def ValuesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Int32Vector
|
|
def ValuesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def Int32VectorStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def Int32VectorAddValues(builder, values):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(values), 0)
|
|
|
|
def Int32VectorStartValuesVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def Int32VectorEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class Int32VectorT(object):
|
|
|
|
# Int32VectorT
|
|
def __init__(self):
|
|
self.values = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
int32Vector = Int32Vector()
|
|
int32Vector.Init(buf, pos)
|
|
return cls.InitFromObj(int32Vector)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, int32Vector):
|
|
x = Int32VectorT()
|
|
x._UnPack(int32Vector)
|
|
return x
|
|
|
|
# Int32VectorT
|
|
def _UnPack(self, int32Vector):
|
|
if int32Vector is None:
|
|
return
|
|
if not int32Vector.ValuesIsNone():
|
|
if np is None:
|
|
self.values = []
|
|
for i in range(int32Vector.ValuesLength()):
|
|
self.values.append(int32Vector.Values(i))
|
|
else:
|
|
self.values = int32Vector.ValuesAsNumpy()
|
|
|
|
# Int32VectorT
|
|
def Pack(self, builder):
|
|
if self.values is not None:
|
|
if np is not None and type(self.values) is np.ndarray:
|
|
values = builder.CreateNumpyVector(self.values)
|
|
else:
|
|
Int32VectorStartValuesVector(builder, len(self.values))
|
|
for i in reversed(range(len(self.values))):
|
|
builder.PrependInt32(self.values[i])
|
|
values = builder.EndVector()
|
|
Int32VectorStart(builder)
|
|
if self.values is not None:
|
|
Int32VectorAddValues(builder, values)
|
|
int32Vector = Int32VectorEnd(builder)
|
|
return int32Vector
|
|
|
|
|
|
class Uint16Vector(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Uint16Vector()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUint16Vector(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Uint16VectorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Uint16Vector
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Uint16Vector
|
|
def Values(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.Uint16Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 2))
|
|
return 0
|
|
|
|
# Uint16Vector
|
|
def ValuesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint16Flags, o)
|
|
return 0
|
|
|
|
# Uint16Vector
|
|
def ValuesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Uint16Vector
|
|
def ValuesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def Uint16VectorStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def Uint16VectorAddValues(builder, values):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(values), 0)
|
|
|
|
def Uint16VectorStartValuesVector(builder, numElems):
|
|
return builder.StartVector(2, numElems, 2)
|
|
|
|
def Uint16VectorEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class Uint16VectorT(object):
|
|
|
|
# Uint16VectorT
|
|
def __init__(self):
|
|
self.values = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
uint16Vector = Uint16Vector()
|
|
uint16Vector.Init(buf, pos)
|
|
return cls.InitFromObj(uint16Vector)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, uint16Vector):
|
|
x = Uint16VectorT()
|
|
x._UnPack(uint16Vector)
|
|
return x
|
|
|
|
# Uint16VectorT
|
|
def _UnPack(self, uint16Vector):
|
|
if uint16Vector is None:
|
|
return
|
|
if not uint16Vector.ValuesIsNone():
|
|
if np is None:
|
|
self.values = []
|
|
for i in range(uint16Vector.ValuesLength()):
|
|
self.values.append(uint16Vector.Values(i))
|
|
else:
|
|
self.values = uint16Vector.ValuesAsNumpy()
|
|
|
|
# Uint16VectorT
|
|
def Pack(self, builder):
|
|
if self.values is not None:
|
|
if np is not None and type(self.values) is np.ndarray:
|
|
values = builder.CreateNumpyVector(self.values)
|
|
else:
|
|
Uint16VectorStartValuesVector(builder, len(self.values))
|
|
for i in reversed(range(len(self.values))):
|
|
builder.PrependUint16(self.values[i])
|
|
values = builder.EndVector()
|
|
Uint16VectorStart(builder)
|
|
if self.values is not None:
|
|
Uint16VectorAddValues(builder, values)
|
|
uint16Vector = Uint16VectorEnd(builder)
|
|
return uint16Vector
|
|
|
|
|
|
class Uint8Vector(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Uint8Vector()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUint8Vector(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Uint8VectorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Uint8Vector
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Uint8Vector
|
|
def Values(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.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# Uint8Vector
|
|
def ValuesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o)
|
|
return 0
|
|
|
|
# Uint8Vector
|
|
def ValuesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Uint8Vector
|
|
def ValuesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def Uint8VectorStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def Uint8VectorAddValues(builder, values):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(values), 0)
|
|
|
|
def Uint8VectorStartValuesVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def Uint8VectorEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class Uint8VectorT(object):
|
|
|
|
# Uint8VectorT
|
|
def __init__(self):
|
|
self.values = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
uint8Vector = Uint8Vector()
|
|
uint8Vector.Init(buf, pos)
|
|
return cls.InitFromObj(uint8Vector)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, uint8Vector):
|
|
x = Uint8VectorT()
|
|
x._UnPack(uint8Vector)
|
|
return x
|
|
|
|
# Uint8VectorT
|
|
def _UnPack(self, uint8Vector):
|
|
if uint8Vector is None:
|
|
return
|
|
if not uint8Vector.ValuesIsNone():
|
|
if np is None:
|
|
self.values = []
|
|
for i in range(uint8Vector.ValuesLength()):
|
|
self.values.append(uint8Vector.Values(i))
|
|
else:
|
|
self.values = uint8Vector.ValuesAsNumpy()
|
|
|
|
# Uint8VectorT
|
|
def Pack(self, builder):
|
|
if self.values is not None:
|
|
if np is not None and type(self.values) is np.ndarray:
|
|
values = builder.CreateNumpyVector(self.values)
|
|
else:
|
|
Uint8VectorStartValuesVector(builder, len(self.values))
|
|
for i in reversed(range(len(self.values))):
|
|
builder.PrependUint8(self.values[i])
|
|
values = builder.EndVector()
|
|
Uint8VectorStart(builder)
|
|
if self.values is not None:
|
|
Uint8VectorAddValues(builder, values)
|
|
uint8Vector = Uint8VectorEnd(builder)
|
|
return uint8Vector
|
|
|
|
|
|
class DimensionMetadata(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DimensionMetadata()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDimensionMetadata(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DimensionMetadataBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DimensionMetadata
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# DimensionMetadata
|
|
def Format(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DimensionMetadata
|
|
def DenseSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DimensionMetadata
|
|
def ArraySegmentsType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DimensionMetadata
|
|
def ArraySegments(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
from flatbuffers.table import Table
|
|
obj = Table(bytearray(), 0)
|
|
self._tab.Union(obj, o)
|
|
return obj
|
|
return None
|
|
|
|
# DimensionMetadata
|
|
def ArrayIndicesType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DimensionMetadata
|
|
def ArrayIndices(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
from flatbuffers.table import Table
|
|
obj = Table(bytearray(), 0)
|
|
self._tab.Union(obj, o)
|
|
return obj
|
|
return None
|
|
|
|
def DimensionMetadataStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def DimensionMetadataAddFormat(builder, format):
|
|
builder.PrependInt8Slot(0, format, 0)
|
|
|
|
def DimensionMetadataAddDenseSize(builder, denseSize):
|
|
builder.PrependInt32Slot(1, denseSize, 0)
|
|
|
|
def DimensionMetadataAddArraySegmentsType(builder, arraySegmentsType):
|
|
builder.PrependUint8Slot(2, arraySegmentsType, 0)
|
|
|
|
def DimensionMetadataAddArraySegments(builder, arraySegments):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(arraySegments), 0)
|
|
|
|
def DimensionMetadataAddArrayIndicesType(builder, arrayIndicesType):
|
|
builder.PrependUint8Slot(4, arrayIndicesType, 0)
|
|
|
|
def DimensionMetadataAddArrayIndices(builder, arrayIndices):
|
|
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(arrayIndices), 0)
|
|
|
|
def DimensionMetadataEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import Union
|
|
except:
|
|
pass
|
|
|
|
class DimensionMetadataT(object):
|
|
|
|
# DimensionMetadataT
|
|
def __init__(self):
|
|
self.format = 0 # type: int
|
|
self.denseSize = 0 # type: int
|
|
self.arraySegmentsType = 0 # type: int
|
|
self.arraySegments = None # type: Union[None, Int32VectorT, Uint16VectorT, Uint8VectorT]
|
|
self.arrayIndicesType = 0 # type: int
|
|
self.arrayIndices = None # type: Union[None, Int32VectorT, Uint16VectorT, Uint8VectorT]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
dimensionMetadata = DimensionMetadata()
|
|
dimensionMetadata.Init(buf, pos)
|
|
return cls.InitFromObj(dimensionMetadata)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, dimensionMetadata):
|
|
x = DimensionMetadataT()
|
|
x._UnPack(dimensionMetadata)
|
|
return x
|
|
|
|
# DimensionMetadataT
|
|
def _UnPack(self, dimensionMetadata):
|
|
if dimensionMetadata is None:
|
|
return
|
|
self.format = dimensionMetadata.Format()
|
|
self.denseSize = dimensionMetadata.DenseSize()
|
|
self.arraySegmentsType = dimensionMetadata.ArraySegmentsType()
|
|
self.arraySegments = SparseIndexVectorCreator(self.arraySegmentsType, dimensionMetadata.ArraySegments())
|
|
self.arrayIndicesType = dimensionMetadata.ArrayIndicesType()
|
|
self.arrayIndices = SparseIndexVectorCreator(self.arrayIndicesType, dimensionMetadata.ArrayIndices())
|
|
|
|
# DimensionMetadataT
|
|
def Pack(self, builder):
|
|
if self.arraySegments is not None:
|
|
arraySegments = self.arraySegments.Pack(builder)
|
|
if self.arrayIndices is not None:
|
|
arrayIndices = self.arrayIndices.Pack(builder)
|
|
DimensionMetadataStart(builder)
|
|
DimensionMetadataAddFormat(builder, self.format)
|
|
DimensionMetadataAddDenseSize(builder, self.denseSize)
|
|
DimensionMetadataAddArraySegmentsType(builder, self.arraySegmentsType)
|
|
if self.arraySegments is not None:
|
|
DimensionMetadataAddArraySegments(builder, arraySegments)
|
|
DimensionMetadataAddArrayIndicesType(builder, self.arrayIndicesType)
|
|
if self.arrayIndices is not None:
|
|
DimensionMetadataAddArrayIndices(builder, arrayIndices)
|
|
dimensionMetadata = DimensionMetadataEnd(builder)
|
|
return dimensionMetadata
|
|
|
|
|
|
class SparsityParameters(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SparsityParameters()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSparsityParameters(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SparsityParametersBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SparsityParameters
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SparsityParameters
|
|
def TraversalOrder(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def TraversalOrderAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def TraversalOrderLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def TraversalOrderIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# SparsityParameters
|
|
def BlockMap(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def BlockMapAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def BlockMapLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def BlockMapIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# SparsityParameters
|
|
def DimMetadata(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = DimensionMetadata()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# SparsityParameters
|
|
def DimMetadataLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SparsityParameters
|
|
def DimMetadataIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
def SparsityParametersStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def SparsityParametersAddTraversalOrder(builder, traversalOrder):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(traversalOrder), 0)
|
|
|
|
def SparsityParametersStartTraversalOrderVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SparsityParametersAddBlockMap(builder, blockMap):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(blockMap), 0)
|
|
|
|
def SparsityParametersStartBlockMapVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SparsityParametersAddDimMetadata(builder, dimMetadata):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(dimMetadata), 0)
|
|
|
|
def SparsityParametersStartDimMetadataVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SparsityParametersEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class SparsityParametersT(object):
|
|
|
|
# SparsityParametersT
|
|
def __init__(self):
|
|
self.traversalOrder = None # type: List[int]
|
|
self.blockMap = None # type: List[int]
|
|
self.dimMetadata = None # type: List[DimensionMetadataT]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
sparsityParameters = SparsityParameters()
|
|
sparsityParameters.Init(buf, pos)
|
|
return cls.InitFromObj(sparsityParameters)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, sparsityParameters):
|
|
x = SparsityParametersT()
|
|
x._UnPack(sparsityParameters)
|
|
return x
|
|
|
|
# SparsityParametersT
|
|
def _UnPack(self, sparsityParameters):
|
|
if sparsityParameters is None:
|
|
return
|
|
if not sparsityParameters.TraversalOrderIsNone():
|
|
if np is None:
|
|
self.traversalOrder = []
|
|
for i in range(sparsityParameters.TraversalOrderLength()):
|
|
self.traversalOrder.append(sparsityParameters.TraversalOrder(i))
|
|
else:
|
|
self.traversalOrder = sparsityParameters.TraversalOrderAsNumpy()
|
|
if not sparsityParameters.BlockMapIsNone():
|
|
if np is None:
|
|
self.blockMap = []
|
|
for i in range(sparsityParameters.BlockMapLength()):
|
|
self.blockMap.append(sparsityParameters.BlockMap(i))
|
|
else:
|
|
self.blockMap = sparsityParameters.BlockMapAsNumpy()
|
|
if not sparsityParameters.DimMetadataIsNone():
|
|
self.dimMetadata = []
|
|
for i in range(sparsityParameters.DimMetadataLength()):
|
|
if sparsityParameters.DimMetadata(i) is None:
|
|
self.dimMetadata.append(None)
|
|
else:
|
|
dimensionMetadata_ = DimensionMetadataT.InitFromObj(sparsityParameters.DimMetadata(i))
|
|
self.dimMetadata.append(dimensionMetadata_)
|
|
|
|
# SparsityParametersT
|
|
def Pack(self, builder):
|
|
if self.traversalOrder is not None:
|
|
if np is not None and type(self.traversalOrder) is np.ndarray:
|
|
traversalOrder = builder.CreateNumpyVector(self.traversalOrder)
|
|
else:
|
|
SparsityParametersStartTraversalOrderVector(builder, len(self.traversalOrder))
|
|
for i in reversed(range(len(self.traversalOrder))):
|
|
builder.PrependInt32(self.traversalOrder[i])
|
|
traversalOrder = builder.EndVector()
|
|
if self.blockMap is not None:
|
|
if np is not None and type(self.blockMap) is np.ndarray:
|
|
blockMap = builder.CreateNumpyVector(self.blockMap)
|
|
else:
|
|
SparsityParametersStartBlockMapVector(builder, len(self.blockMap))
|
|
for i in reversed(range(len(self.blockMap))):
|
|
builder.PrependInt32(self.blockMap[i])
|
|
blockMap = builder.EndVector()
|
|
if self.dimMetadata is not None:
|
|
dimMetadatalist = []
|
|
for i in range(len(self.dimMetadata)):
|
|
dimMetadatalist.append(self.dimMetadata[i].Pack(builder))
|
|
SparsityParametersStartDimMetadataVector(builder, len(self.dimMetadata))
|
|
for i in reversed(range(len(self.dimMetadata))):
|
|
builder.PrependUOffsetTRelative(dimMetadatalist[i])
|
|
dimMetadata = builder.EndVector()
|
|
SparsityParametersStart(builder)
|
|
if self.traversalOrder is not None:
|
|
SparsityParametersAddTraversalOrder(builder, traversalOrder)
|
|
if self.blockMap is not None:
|
|
SparsityParametersAddBlockMap(builder, blockMap)
|
|
if self.dimMetadata is not None:
|
|
SparsityParametersAddDimMetadata(builder, dimMetadata)
|
|
sparsityParameters = SparsityParametersEnd(builder)
|
|
return sparsityParameters
|
|
|
|
|
|
class VariantSubType(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = VariantSubType()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsVariantSubType(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def VariantSubTypeBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# VariantSubType
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# VariantSubType
|
|
def Shape(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# VariantSubType
|
|
def ShapeAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# VariantSubType
|
|
def ShapeLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# VariantSubType
|
|
def ShapeIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# VariantSubType
|
|
def Type(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# VariantSubType
|
|
def HasRank(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def VariantSubTypeStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def VariantSubTypeAddShape(builder, shape):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0)
|
|
|
|
def VariantSubTypeStartShapeVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def VariantSubTypeAddType(builder, type):
|
|
builder.PrependInt8Slot(1, type, 0)
|
|
|
|
def VariantSubTypeAddHasRank(builder, hasRank):
|
|
builder.PrependBoolSlot(2, hasRank, 0)
|
|
|
|
def VariantSubTypeEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class VariantSubTypeT(object):
|
|
|
|
# VariantSubTypeT
|
|
def __init__(self):
|
|
self.shape = None # type: List[int]
|
|
self.type = 0 # type: int
|
|
self.hasRank = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
variantSubType = VariantSubType()
|
|
variantSubType.Init(buf, pos)
|
|
return cls.InitFromObj(variantSubType)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, variantSubType):
|
|
x = VariantSubTypeT()
|
|
x._UnPack(variantSubType)
|
|
return x
|
|
|
|
# VariantSubTypeT
|
|
def _UnPack(self, variantSubType):
|
|
if variantSubType is None:
|
|
return
|
|
if not variantSubType.ShapeIsNone():
|
|
if np is None:
|
|
self.shape = []
|
|
for i in range(variantSubType.ShapeLength()):
|
|
self.shape.append(variantSubType.Shape(i))
|
|
else:
|
|
self.shape = variantSubType.ShapeAsNumpy()
|
|
self.type = variantSubType.Type()
|
|
self.hasRank = variantSubType.HasRank()
|
|
|
|
# VariantSubTypeT
|
|
def Pack(self, builder):
|
|
if self.shape is not None:
|
|
if np is not None and type(self.shape) is np.ndarray:
|
|
shape = builder.CreateNumpyVector(self.shape)
|
|
else:
|
|
VariantSubTypeStartShapeVector(builder, len(self.shape))
|
|
for i in reversed(range(len(self.shape))):
|
|
builder.PrependInt32(self.shape[i])
|
|
shape = builder.EndVector()
|
|
VariantSubTypeStart(builder)
|
|
if self.shape is not None:
|
|
VariantSubTypeAddShape(builder, shape)
|
|
VariantSubTypeAddType(builder, self.type)
|
|
VariantSubTypeAddHasRank(builder, self.hasRank)
|
|
variantSubType = VariantSubTypeEnd(builder)
|
|
return variantSubType
|
|
|
|
|
|
class Tensor(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Tensor()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTensor(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TensorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Tensor
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Tensor
|
|
def Shape(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# Tensor
|
|
def Type(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Tensor
|
|
def Buffer(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Tensor
|
|
def Name(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# Tensor
|
|
def Quantization(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
x = self._tab.Indirect(o + self._tab.Pos)
|
|
obj = QuantizationParameters()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Tensor
|
|
def IsVariable(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# Tensor
|
|
def Sparsity(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
x = self._tab.Indirect(o + self._tab.Pos)
|
|
obj = SparsityParameters()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Tensor
|
|
def ShapeSignature(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeSignatureAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeSignatureLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Tensor
|
|
def ShapeSignatureIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
return o == 0
|
|
|
|
# Tensor
|
|
def HasRank(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# Tensor
|
|
def VariantTensors(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = VariantSubType()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Tensor
|
|
def VariantTensorsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Tensor
|
|
def VariantTensorsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
return o == 0
|
|
|
|
def TensorStart(builder):
|
|
builder.StartObject(10)
|
|
|
|
def TensorAddShape(builder, shape):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0)
|
|
|
|
def TensorStartShapeVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def TensorAddType(builder, type):
|
|
builder.PrependInt8Slot(1, type, 0)
|
|
|
|
def TensorAddBuffer(builder, buffer):
|
|
builder.PrependUint32Slot(2, buffer, 0)
|
|
|
|
def TensorAddName(builder, name):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
|
|
|
|
def TensorAddQuantization(builder, quantization):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(quantization), 0)
|
|
|
|
def TensorAddIsVariable(builder, isVariable):
|
|
builder.PrependBoolSlot(5, isVariable, 0)
|
|
|
|
def TensorAddSparsity(builder, sparsity):
|
|
builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(sparsity), 0)
|
|
|
|
def TensorAddShapeSignature(builder, shapeSignature):
|
|
builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(shapeSignature), 0)
|
|
|
|
def TensorStartShapeSignatureVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def TensorAddHasRank(builder, hasRank):
|
|
builder.PrependBoolSlot(8, hasRank, 0)
|
|
|
|
def TensorAddVariantTensors(builder, variantTensors):
|
|
builder.PrependUOffsetTRelativeSlot(9, flatbuffers.number_types.UOffsetTFlags.py_type(variantTensors), 0)
|
|
|
|
def TensorStartVariantTensorsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def TensorEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List, Optional
|
|
except:
|
|
pass
|
|
|
|
class TensorT(object):
|
|
|
|
# TensorT
|
|
def __init__(self):
|
|
self.shape = None # type: List[int]
|
|
self.type = 0 # type: int
|
|
self.buffer = 0 # type: int
|
|
self.name = None # type: str
|
|
self.quantization = None # type: Optional[QuantizationParametersT]
|
|
self.isVariable = False # type: bool
|
|
self.sparsity = None # type: Optional[SparsityParametersT]
|
|
self.shapeSignature = None # type: List[int]
|
|
self.hasRank = False # type: bool
|
|
self.variantTensors = None # type: List[VariantSubTypeT]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
tensor = Tensor()
|
|
tensor.Init(buf, pos)
|
|
return cls.InitFromObj(tensor)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, tensor):
|
|
x = TensorT()
|
|
x._UnPack(tensor)
|
|
return x
|
|
|
|
# TensorT
|
|
def _UnPack(self, tensor):
|
|
if tensor is None:
|
|
return
|
|
if not tensor.ShapeIsNone():
|
|
if np is None:
|
|
self.shape = []
|
|
for i in range(tensor.ShapeLength()):
|
|
self.shape.append(tensor.Shape(i))
|
|
else:
|
|
self.shape = tensor.ShapeAsNumpy()
|
|
self.type = tensor.Type()
|
|
self.buffer = tensor.Buffer()
|
|
self.name = tensor.Name()
|
|
if tensor.Quantization() is not None:
|
|
self.quantization = QuantizationParametersT.InitFromObj(tensor.Quantization())
|
|
self.isVariable = tensor.IsVariable()
|
|
if tensor.Sparsity() is not None:
|
|
self.sparsity = SparsityParametersT.InitFromObj(tensor.Sparsity())
|
|
if not tensor.ShapeSignatureIsNone():
|
|
if np is None:
|
|
self.shapeSignature = []
|
|
for i in range(tensor.ShapeSignatureLength()):
|
|
self.shapeSignature.append(tensor.ShapeSignature(i))
|
|
else:
|
|
self.shapeSignature = tensor.ShapeSignatureAsNumpy()
|
|
self.hasRank = tensor.HasRank()
|
|
if not tensor.VariantTensorsIsNone():
|
|
self.variantTensors = []
|
|
for i in range(tensor.VariantTensorsLength()):
|
|
if tensor.VariantTensors(i) is None:
|
|
self.variantTensors.append(None)
|
|
else:
|
|
variantSubType_ = VariantSubTypeT.InitFromObj(tensor.VariantTensors(i))
|
|
self.variantTensors.append(variantSubType_)
|
|
|
|
# TensorT
|
|
def Pack(self, builder):
|
|
if self.shape is not None:
|
|
if np is not None and type(self.shape) is np.ndarray:
|
|
shape = builder.CreateNumpyVector(self.shape)
|
|
else:
|
|
TensorStartShapeVector(builder, len(self.shape))
|
|
for i in reversed(range(len(self.shape))):
|
|
builder.PrependInt32(self.shape[i])
|
|
shape = builder.EndVector()
|
|
if self.name is not None:
|
|
name = builder.CreateString(self.name)
|
|
if self.quantization is not None:
|
|
quantization = self.quantization.Pack(builder)
|
|
if self.sparsity is not None:
|
|
sparsity = self.sparsity.Pack(builder)
|
|
if self.shapeSignature is not None:
|
|
if np is not None and type(self.shapeSignature) is np.ndarray:
|
|
shapeSignature = builder.CreateNumpyVector(self.shapeSignature)
|
|
else:
|
|
TensorStartShapeSignatureVector(builder, len(self.shapeSignature))
|
|
for i in reversed(range(len(self.shapeSignature))):
|
|
builder.PrependInt32(self.shapeSignature[i])
|
|
shapeSignature = builder.EndVector()
|
|
if self.variantTensors is not None:
|
|
variantTensorslist = []
|
|
for i in range(len(self.variantTensors)):
|
|
variantTensorslist.append(self.variantTensors[i].Pack(builder))
|
|
TensorStartVariantTensorsVector(builder, len(self.variantTensors))
|
|
for i in reversed(range(len(self.variantTensors))):
|
|
builder.PrependUOffsetTRelative(variantTensorslist[i])
|
|
variantTensors = builder.EndVector()
|
|
TensorStart(builder)
|
|
if self.shape is not None:
|
|
TensorAddShape(builder, shape)
|
|
TensorAddType(builder, self.type)
|
|
TensorAddBuffer(builder, self.buffer)
|
|
if self.name is not None:
|
|
TensorAddName(builder, name)
|
|
if self.quantization is not None:
|
|
TensorAddQuantization(builder, quantization)
|
|
TensorAddIsVariable(builder, self.isVariable)
|
|
if self.sparsity is not None:
|
|
TensorAddSparsity(builder, sparsity)
|
|
if self.shapeSignature is not None:
|
|
TensorAddShapeSignature(builder, shapeSignature)
|
|
TensorAddHasRank(builder, self.hasRank)
|
|
if self.variantTensors is not None:
|
|
TensorAddVariantTensors(builder, variantTensors)
|
|
tensor = TensorEnd(builder)
|
|
return tensor
|
|
|
|
|
|
class StablehloGatherOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloGatherOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloGatherOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloGatherOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloGatherOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloGatherOptions
|
|
def OffsetDims(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
|
|
|
|
# StablehloGatherOptions
|
|
def OffsetDimsAsNumpy(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
|
|
|
|
# StablehloGatherOptions
|
|
def OffsetDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def OffsetDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloGatherOptions
|
|
def CollapsedSliceDims(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
|
|
|
|
# StablehloGatherOptions
|
|
def CollapsedSliceDimsAsNumpy(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
|
|
|
|
# StablehloGatherOptions
|
|
def CollapsedSliceDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def CollapsedSliceDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloGatherOptions
|
|
def StartIndexMap(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
|
|
|
|
# StablehloGatherOptions
|
|
def StartIndexMapAsNumpy(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
|
|
|
|
# StablehloGatherOptions
|
|
def StartIndexMapLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def StartIndexMapIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# StablehloGatherOptions
|
|
def IndexVectorDim(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def SliceSizes(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.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8))
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def SliceSizesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def SliceSizesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloGatherOptions
|
|
def SliceSizesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
return o == 0
|
|
|
|
# StablehloGatherOptions
|
|
def IndicesAreSorted(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def StablehloGatherOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def StablehloGatherOptionsAddOffsetDims(builder, offsetDims):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(offsetDims), 0)
|
|
|
|
def StablehloGatherOptionsStartOffsetDimsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloGatherOptionsAddCollapsedSliceDims(builder, collapsedSliceDims):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(collapsedSliceDims), 0)
|
|
|
|
def StablehloGatherOptionsStartCollapsedSliceDimsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloGatherOptionsAddStartIndexMap(builder, startIndexMap):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(startIndexMap), 0)
|
|
|
|
def StablehloGatherOptionsStartStartIndexMapVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloGatherOptionsAddIndexVectorDim(builder, indexVectorDim):
|
|
builder.PrependInt64Slot(3, indexVectorDim, 0)
|
|
|
|
def StablehloGatherOptionsAddSliceSizes(builder, sliceSizes):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(sliceSizes), 0)
|
|
|
|
def StablehloGatherOptionsStartSliceSizesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloGatherOptionsAddIndicesAreSorted(builder, indicesAreSorted):
|
|
builder.PrependBoolSlot(5, indicesAreSorted, 0)
|
|
|
|
def StablehloGatherOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloGatherOptionsT(object):
|
|
|
|
# StablehloGatherOptionsT
|
|
def __init__(self):
|
|
self.offsetDims = None # type: List[int]
|
|
self.collapsedSliceDims = None # type: List[int]
|
|
self.startIndexMap = None # type: List[int]
|
|
self.indexVectorDim = 0 # type: int
|
|
self.sliceSizes = None # type: List[int]
|
|
self.indicesAreSorted = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloGatherOptions = StablehloGatherOptions()
|
|
stablehloGatherOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloGatherOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloGatherOptions):
|
|
x = StablehloGatherOptionsT()
|
|
x._UnPack(stablehloGatherOptions)
|
|
return x
|
|
|
|
# StablehloGatherOptionsT
|
|
def _UnPack(self, stablehloGatherOptions):
|
|
if stablehloGatherOptions is None:
|
|
return
|
|
if not stablehloGatherOptions.OffsetDimsIsNone():
|
|
if np is None:
|
|
self.offsetDims = []
|
|
for i in range(stablehloGatherOptions.OffsetDimsLength()):
|
|
self.offsetDims.append(stablehloGatherOptions.OffsetDims(i))
|
|
else:
|
|
self.offsetDims = stablehloGatherOptions.OffsetDimsAsNumpy()
|
|
if not stablehloGatherOptions.CollapsedSliceDimsIsNone():
|
|
if np is None:
|
|
self.collapsedSliceDims = []
|
|
for i in range(stablehloGatherOptions.CollapsedSliceDimsLength()):
|
|
self.collapsedSliceDims.append(stablehloGatherOptions.CollapsedSliceDims(i))
|
|
else:
|
|
self.collapsedSliceDims = stablehloGatherOptions.CollapsedSliceDimsAsNumpy()
|
|
if not stablehloGatherOptions.StartIndexMapIsNone():
|
|
if np is None:
|
|
self.startIndexMap = []
|
|
for i in range(stablehloGatherOptions.StartIndexMapLength()):
|
|
self.startIndexMap.append(stablehloGatherOptions.StartIndexMap(i))
|
|
else:
|
|
self.startIndexMap = stablehloGatherOptions.StartIndexMapAsNumpy()
|
|
self.indexVectorDim = stablehloGatherOptions.IndexVectorDim()
|
|
if not stablehloGatherOptions.SliceSizesIsNone():
|
|
if np is None:
|
|
self.sliceSizes = []
|
|
for i in range(stablehloGatherOptions.SliceSizesLength()):
|
|
self.sliceSizes.append(stablehloGatherOptions.SliceSizes(i))
|
|
else:
|
|
self.sliceSizes = stablehloGatherOptions.SliceSizesAsNumpy()
|
|
self.indicesAreSorted = stablehloGatherOptions.IndicesAreSorted()
|
|
|
|
# StablehloGatherOptionsT
|
|
def Pack(self, builder):
|
|
if self.offsetDims is not None:
|
|
if np is not None and type(self.offsetDims) is np.ndarray:
|
|
offsetDims = builder.CreateNumpyVector(self.offsetDims)
|
|
else:
|
|
StablehloGatherOptionsStartOffsetDimsVector(builder, len(self.offsetDims))
|
|
for i in reversed(range(len(self.offsetDims))):
|
|
builder.PrependInt64(self.offsetDims[i])
|
|
offsetDims = builder.EndVector()
|
|
if self.collapsedSliceDims is not None:
|
|
if np is not None and type(self.collapsedSliceDims) is np.ndarray:
|
|
collapsedSliceDims = builder.CreateNumpyVector(self.collapsedSliceDims)
|
|
else:
|
|
StablehloGatherOptionsStartCollapsedSliceDimsVector(builder, len(self.collapsedSliceDims))
|
|
for i in reversed(range(len(self.collapsedSliceDims))):
|
|
builder.PrependInt64(self.collapsedSliceDims[i])
|
|
collapsedSliceDims = builder.EndVector()
|
|
if self.startIndexMap is not None:
|
|
if np is not None and type(self.startIndexMap) is np.ndarray:
|
|
startIndexMap = builder.CreateNumpyVector(self.startIndexMap)
|
|
else:
|
|
StablehloGatherOptionsStartStartIndexMapVector(builder, len(self.startIndexMap))
|
|
for i in reversed(range(len(self.startIndexMap))):
|
|
builder.PrependInt64(self.startIndexMap[i])
|
|
startIndexMap = builder.EndVector()
|
|
if self.sliceSizes is not None:
|
|
if np is not None and type(self.sliceSizes) is np.ndarray:
|
|
sliceSizes = builder.CreateNumpyVector(self.sliceSizes)
|
|
else:
|
|
StablehloGatherOptionsStartSliceSizesVector(builder, len(self.sliceSizes))
|
|
for i in reversed(range(len(self.sliceSizes))):
|
|
builder.PrependInt64(self.sliceSizes[i])
|
|
sliceSizes = builder.EndVector()
|
|
StablehloGatherOptionsStart(builder)
|
|
if self.offsetDims is not None:
|
|
StablehloGatherOptionsAddOffsetDims(builder, offsetDims)
|
|
if self.collapsedSliceDims is not None:
|
|
StablehloGatherOptionsAddCollapsedSliceDims(builder, collapsedSliceDims)
|
|
if self.startIndexMap is not None:
|
|
StablehloGatherOptionsAddStartIndexMap(builder, startIndexMap)
|
|
StablehloGatherOptionsAddIndexVectorDim(builder, self.indexVectorDim)
|
|
if self.sliceSizes is not None:
|
|
StablehloGatherOptionsAddSliceSizes(builder, sliceSizes)
|
|
StablehloGatherOptionsAddIndicesAreSorted(builder, self.indicesAreSorted)
|
|
stablehloGatherOptions = StablehloGatherOptionsEnd(builder)
|
|
return stablehloGatherOptions
|
|
|
|
|
|
class StablehloTransposeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloTransposeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloTransposeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloTransposeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloTransposeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloTransposeOptions
|
|
def Permutation(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
|
|
|
|
# StablehloTransposeOptions
|
|
def PermutationAsNumpy(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
|
|
|
|
# StablehloTransposeOptions
|
|
def PermutationLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloTransposeOptions
|
|
def PermutationIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def StablehloTransposeOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloTransposeOptionsAddPermutation(builder, permutation):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(permutation), 0)
|
|
|
|
def StablehloTransposeOptionsStartPermutationVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloTransposeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloTransposeOptionsT(object):
|
|
|
|
# StablehloTransposeOptionsT
|
|
def __init__(self):
|
|
self.permutation = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloTransposeOptions = StablehloTransposeOptions()
|
|
stablehloTransposeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloTransposeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloTransposeOptions):
|
|
x = StablehloTransposeOptionsT()
|
|
x._UnPack(stablehloTransposeOptions)
|
|
return x
|
|
|
|
# StablehloTransposeOptionsT
|
|
def _UnPack(self, stablehloTransposeOptions):
|
|
if stablehloTransposeOptions is None:
|
|
return
|
|
if not stablehloTransposeOptions.PermutationIsNone():
|
|
if np is None:
|
|
self.permutation = []
|
|
for i in range(stablehloTransposeOptions.PermutationLength()):
|
|
self.permutation.append(stablehloTransposeOptions.Permutation(i))
|
|
else:
|
|
self.permutation = stablehloTransposeOptions.PermutationAsNumpy()
|
|
|
|
# StablehloTransposeOptionsT
|
|
def Pack(self, builder):
|
|
if self.permutation is not None:
|
|
if np is not None and type(self.permutation) is np.ndarray:
|
|
permutation = builder.CreateNumpyVector(self.permutation)
|
|
else:
|
|
StablehloTransposeOptionsStartPermutationVector(builder, len(self.permutation))
|
|
for i in reversed(range(len(self.permutation))):
|
|
builder.PrependInt64(self.permutation[i])
|
|
permutation = builder.EndVector()
|
|
StablehloTransposeOptionsStart(builder)
|
|
if self.permutation is not None:
|
|
StablehloTransposeOptionsAddPermutation(builder, permutation)
|
|
stablehloTransposeOptions = StablehloTransposeOptionsEnd(builder)
|
|
return stablehloTransposeOptions
|
|
|
|
|
|
class StablehloDotGeneralOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloDotGeneralOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloDotGeneralOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloDotGeneralOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloDotGeneralOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsBatchingDimensions(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsBatchingDimensionsAsNumpy(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsBatchingDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsBatchingDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsBatchingDimensions(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsBatchingDimensionsAsNumpy(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsBatchingDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsBatchingDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsContractingDimensions(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsContractingDimensionsAsNumpy(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsContractingDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def LhsContractingDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsContractingDimensions(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsContractingDimensionsAsNumpy(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
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsContractingDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def RhsContractingDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
return o == 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def PrecisionConfig(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.Uint32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def PrecisionConfigAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint32Flags, o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def PrecisionConfigLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDotGeneralOptions
|
|
def PrecisionConfigIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
return o == 0
|
|
|
|
def StablehloDotGeneralOptionsStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def StablehloDotGeneralOptionsAddLhsBatchingDimensions(builder, lhsBatchingDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(lhsBatchingDimensions), 0)
|
|
|
|
def StablehloDotGeneralOptionsStartLhsBatchingDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloDotGeneralOptionsAddRhsBatchingDimensions(builder, rhsBatchingDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(rhsBatchingDimensions), 0)
|
|
|
|
def StablehloDotGeneralOptionsStartRhsBatchingDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloDotGeneralOptionsAddLhsContractingDimensions(builder, lhsContractingDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(lhsContractingDimensions), 0)
|
|
|
|
def StablehloDotGeneralOptionsStartLhsContractingDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloDotGeneralOptionsAddRhsContractingDimensions(builder, rhsContractingDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(rhsContractingDimensions), 0)
|
|
|
|
def StablehloDotGeneralOptionsStartRhsContractingDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloDotGeneralOptionsAddPrecisionConfig(builder, precisionConfig):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(precisionConfig), 0)
|
|
|
|
def StablehloDotGeneralOptionsStartPrecisionConfigVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def StablehloDotGeneralOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloDotGeneralOptionsT(object):
|
|
|
|
# StablehloDotGeneralOptionsT
|
|
def __init__(self):
|
|
self.lhsBatchingDimensions = None # type: List[int]
|
|
self.rhsBatchingDimensions = None # type: List[int]
|
|
self.lhsContractingDimensions = None # type: List[int]
|
|
self.rhsContractingDimensions = None # type: List[int]
|
|
self.precisionConfig = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloDotGeneralOptions = StablehloDotGeneralOptions()
|
|
stablehloDotGeneralOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloDotGeneralOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloDotGeneralOptions):
|
|
x = StablehloDotGeneralOptionsT()
|
|
x._UnPack(stablehloDotGeneralOptions)
|
|
return x
|
|
|
|
# StablehloDotGeneralOptionsT
|
|
def _UnPack(self, stablehloDotGeneralOptions):
|
|
if stablehloDotGeneralOptions is None:
|
|
return
|
|
if not stablehloDotGeneralOptions.LhsBatchingDimensionsIsNone():
|
|
if np is None:
|
|
self.lhsBatchingDimensions = []
|
|
for i in range(stablehloDotGeneralOptions.LhsBatchingDimensionsLength()):
|
|
self.lhsBatchingDimensions.append(stablehloDotGeneralOptions.LhsBatchingDimensions(i))
|
|
else:
|
|
self.lhsBatchingDimensions = stablehloDotGeneralOptions.LhsBatchingDimensionsAsNumpy()
|
|
if not stablehloDotGeneralOptions.RhsBatchingDimensionsIsNone():
|
|
if np is None:
|
|
self.rhsBatchingDimensions = []
|
|
for i in range(stablehloDotGeneralOptions.RhsBatchingDimensionsLength()):
|
|
self.rhsBatchingDimensions.append(stablehloDotGeneralOptions.RhsBatchingDimensions(i))
|
|
else:
|
|
self.rhsBatchingDimensions = stablehloDotGeneralOptions.RhsBatchingDimensionsAsNumpy()
|
|
if not stablehloDotGeneralOptions.LhsContractingDimensionsIsNone():
|
|
if np is None:
|
|
self.lhsContractingDimensions = []
|
|
for i in range(stablehloDotGeneralOptions.LhsContractingDimensionsLength()):
|
|
self.lhsContractingDimensions.append(stablehloDotGeneralOptions.LhsContractingDimensions(i))
|
|
else:
|
|
self.lhsContractingDimensions = stablehloDotGeneralOptions.LhsContractingDimensionsAsNumpy()
|
|
if not stablehloDotGeneralOptions.RhsContractingDimensionsIsNone():
|
|
if np is None:
|
|
self.rhsContractingDimensions = []
|
|
for i in range(stablehloDotGeneralOptions.RhsContractingDimensionsLength()):
|
|
self.rhsContractingDimensions.append(stablehloDotGeneralOptions.RhsContractingDimensions(i))
|
|
else:
|
|
self.rhsContractingDimensions = stablehloDotGeneralOptions.RhsContractingDimensionsAsNumpy()
|
|
if not stablehloDotGeneralOptions.PrecisionConfigIsNone():
|
|
if np is None:
|
|
self.precisionConfig = []
|
|
for i in range(stablehloDotGeneralOptions.PrecisionConfigLength()):
|
|
self.precisionConfig.append(stablehloDotGeneralOptions.PrecisionConfig(i))
|
|
else:
|
|
self.precisionConfig = stablehloDotGeneralOptions.PrecisionConfigAsNumpy()
|
|
|
|
# StablehloDotGeneralOptionsT
|
|
def Pack(self, builder):
|
|
if self.lhsBatchingDimensions is not None:
|
|
if np is not None and type(self.lhsBatchingDimensions) is np.ndarray:
|
|
lhsBatchingDimensions = builder.CreateNumpyVector(self.lhsBatchingDimensions)
|
|
else:
|
|
StablehloDotGeneralOptionsStartLhsBatchingDimensionsVector(builder, len(self.lhsBatchingDimensions))
|
|
for i in reversed(range(len(self.lhsBatchingDimensions))):
|
|
builder.PrependInt64(self.lhsBatchingDimensions[i])
|
|
lhsBatchingDimensions = builder.EndVector()
|
|
if self.rhsBatchingDimensions is not None:
|
|
if np is not None and type(self.rhsBatchingDimensions) is np.ndarray:
|
|
rhsBatchingDimensions = builder.CreateNumpyVector(self.rhsBatchingDimensions)
|
|
else:
|
|
StablehloDotGeneralOptionsStartRhsBatchingDimensionsVector(builder, len(self.rhsBatchingDimensions))
|
|
for i in reversed(range(len(self.rhsBatchingDimensions))):
|
|
builder.PrependInt64(self.rhsBatchingDimensions[i])
|
|
rhsBatchingDimensions = builder.EndVector()
|
|
if self.lhsContractingDimensions is not None:
|
|
if np is not None and type(self.lhsContractingDimensions) is np.ndarray:
|
|
lhsContractingDimensions = builder.CreateNumpyVector(self.lhsContractingDimensions)
|
|
else:
|
|
StablehloDotGeneralOptionsStartLhsContractingDimensionsVector(builder, len(self.lhsContractingDimensions))
|
|
for i in reversed(range(len(self.lhsContractingDimensions))):
|
|
builder.PrependInt64(self.lhsContractingDimensions[i])
|
|
lhsContractingDimensions = builder.EndVector()
|
|
if self.rhsContractingDimensions is not None:
|
|
if np is not None and type(self.rhsContractingDimensions) is np.ndarray:
|
|
rhsContractingDimensions = builder.CreateNumpyVector(self.rhsContractingDimensions)
|
|
else:
|
|
StablehloDotGeneralOptionsStartRhsContractingDimensionsVector(builder, len(self.rhsContractingDimensions))
|
|
for i in reversed(range(len(self.rhsContractingDimensions))):
|
|
builder.PrependInt64(self.rhsContractingDimensions[i])
|
|
rhsContractingDimensions = builder.EndVector()
|
|
if self.precisionConfig is not None:
|
|
if np is not None and type(self.precisionConfig) is np.ndarray:
|
|
precisionConfig = builder.CreateNumpyVector(self.precisionConfig)
|
|
else:
|
|
StablehloDotGeneralOptionsStartPrecisionConfigVector(builder, len(self.precisionConfig))
|
|
for i in reversed(range(len(self.precisionConfig))):
|
|
builder.PrependUint32(self.precisionConfig[i])
|
|
precisionConfig = builder.EndVector()
|
|
StablehloDotGeneralOptionsStart(builder)
|
|
if self.lhsBatchingDimensions is not None:
|
|
StablehloDotGeneralOptionsAddLhsBatchingDimensions(builder, lhsBatchingDimensions)
|
|
if self.rhsBatchingDimensions is not None:
|
|
StablehloDotGeneralOptionsAddRhsBatchingDimensions(builder, rhsBatchingDimensions)
|
|
if self.lhsContractingDimensions is not None:
|
|
StablehloDotGeneralOptionsAddLhsContractingDimensions(builder, lhsContractingDimensions)
|
|
if self.rhsContractingDimensions is not None:
|
|
StablehloDotGeneralOptionsAddRhsContractingDimensions(builder, rhsContractingDimensions)
|
|
if self.precisionConfig is not None:
|
|
StablehloDotGeneralOptionsAddPrecisionConfig(builder, precisionConfig)
|
|
stablehloDotGeneralOptions = StablehloDotGeneralOptionsEnd(builder)
|
|
return stablehloDotGeneralOptions
|
|
|
|
|
|
class StablehloReduceWindowOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloReduceWindowOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloReduceWindowOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloReduceWindowOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloReduceWindowOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDimensions(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDimensionsAsNumpy(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowStrides(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowStridesAsNumpy(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowStridesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowStridesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def BaseDilations(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def BaseDilationsAsNumpy(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def BaseDilationsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def BaseDilationsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDilations(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDilationsAsNumpy(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
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDilationsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def WindowDilationsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
return o == 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def Padding(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.Int64Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 8))
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def PaddingAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int64Flags, o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def PaddingLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def PaddingIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
return o == 0
|
|
|
|
# StablehloReduceWindowOptions
|
|
def BodySubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloReduceWindowOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def StablehloReduceWindowOptionsAddWindowDimensions(builder, windowDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(windowDimensions), 0)
|
|
|
|
def StablehloReduceWindowOptionsStartWindowDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceWindowOptionsAddWindowStrides(builder, windowStrides):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(windowStrides), 0)
|
|
|
|
def StablehloReduceWindowOptionsStartWindowStridesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceWindowOptionsAddBaseDilations(builder, baseDilations):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(baseDilations), 0)
|
|
|
|
def StablehloReduceWindowOptionsStartBaseDilationsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceWindowOptionsAddWindowDilations(builder, windowDilations):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(windowDilations), 0)
|
|
|
|
def StablehloReduceWindowOptionsStartWindowDilationsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceWindowOptionsAddPadding(builder, padding):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(padding), 0)
|
|
|
|
def StablehloReduceWindowOptionsStartPaddingVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceWindowOptionsAddBodySubgraphIndex(builder, bodySubgraphIndex):
|
|
builder.PrependInt32Slot(5, bodySubgraphIndex, 0)
|
|
|
|
def StablehloReduceWindowOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloReduceWindowOptionsT(object):
|
|
|
|
# StablehloReduceWindowOptionsT
|
|
def __init__(self):
|
|
self.windowDimensions = None # type: List[int]
|
|
self.windowStrides = None # type: List[int]
|
|
self.baseDilations = None # type: List[int]
|
|
self.windowDilations = None # type: List[int]
|
|
self.padding = None # type: List[int]
|
|
self.bodySubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloReduceWindowOptions = StablehloReduceWindowOptions()
|
|
stablehloReduceWindowOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloReduceWindowOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloReduceWindowOptions):
|
|
x = StablehloReduceWindowOptionsT()
|
|
x._UnPack(stablehloReduceWindowOptions)
|
|
return x
|
|
|
|
# StablehloReduceWindowOptionsT
|
|
def _UnPack(self, stablehloReduceWindowOptions):
|
|
if stablehloReduceWindowOptions is None:
|
|
return
|
|
if not stablehloReduceWindowOptions.WindowDimensionsIsNone():
|
|
if np is None:
|
|
self.windowDimensions = []
|
|
for i in range(stablehloReduceWindowOptions.WindowDimensionsLength()):
|
|
self.windowDimensions.append(stablehloReduceWindowOptions.WindowDimensions(i))
|
|
else:
|
|
self.windowDimensions = stablehloReduceWindowOptions.WindowDimensionsAsNumpy()
|
|
if not stablehloReduceWindowOptions.WindowStridesIsNone():
|
|
if np is None:
|
|
self.windowStrides = []
|
|
for i in range(stablehloReduceWindowOptions.WindowStridesLength()):
|
|
self.windowStrides.append(stablehloReduceWindowOptions.WindowStrides(i))
|
|
else:
|
|
self.windowStrides = stablehloReduceWindowOptions.WindowStridesAsNumpy()
|
|
if not stablehloReduceWindowOptions.BaseDilationsIsNone():
|
|
if np is None:
|
|
self.baseDilations = []
|
|
for i in range(stablehloReduceWindowOptions.BaseDilationsLength()):
|
|
self.baseDilations.append(stablehloReduceWindowOptions.BaseDilations(i))
|
|
else:
|
|
self.baseDilations = stablehloReduceWindowOptions.BaseDilationsAsNumpy()
|
|
if not stablehloReduceWindowOptions.WindowDilationsIsNone():
|
|
if np is None:
|
|
self.windowDilations = []
|
|
for i in range(stablehloReduceWindowOptions.WindowDilationsLength()):
|
|
self.windowDilations.append(stablehloReduceWindowOptions.WindowDilations(i))
|
|
else:
|
|
self.windowDilations = stablehloReduceWindowOptions.WindowDilationsAsNumpy()
|
|
if not stablehloReduceWindowOptions.PaddingIsNone():
|
|
if np is None:
|
|
self.padding = []
|
|
for i in range(stablehloReduceWindowOptions.PaddingLength()):
|
|
self.padding.append(stablehloReduceWindowOptions.Padding(i))
|
|
else:
|
|
self.padding = stablehloReduceWindowOptions.PaddingAsNumpy()
|
|
self.bodySubgraphIndex = stablehloReduceWindowOptions.BodySubgraphIndex()
|
|
|
|
# StablehloReduceWindowOptionsT
|
|
def Pack(self, builder):
|
|
if self.windowDimensions is not None:
|
|
if np is not None and type(self.windowDimensions) is np.ndarray:
|
|
windowDimensions = builder.CreateNumpyVector(self.windowDimensions)
|
|
else:
|
|
StablehloReduceWindowOptionsStartWindowDimensionsVector(builder, len(self.windowDimensions))
|
|
for i in reversed(range(len(self.windowDimensions))):
|
|
builder.PrependInt64(self.windowDimensions[i])
|
|
windowDimensions = builder.EndVector()
|
|
if self.windowStrides is not None:
|
|
if np is not None and type(self.windowStrides) is np.ndarray:
|
|
windowStrides = builder.CreateNumpyVector(self.windowStrides)
|
|
else:
|
|
StablehloReduceWindowOptionsStartWindowStridesVector(builder, len(self.windowStrides))
|
|
for i in reversed(range(len(self.windowStrides))):
|
|
builder.PrependInt64(self.windowStrides[i])
|
|
windowStrides = builder.EndVector()
|
|
if self.baseDilations is not None:
|
|
if np is not None and type(self.baseDilations) is np.ndarray:
|
|
baseDilations = builder.CreateNumpyVector(self.baseDilations)
|
|
else:
|
|
StablehloReduceWindowOptionsStartBaseDilationsVector(builder, len(self.baseDilations))
|
|
for i in reversed(range(len(self.baseDilations))):
|
|
builder.PrependInt64(self.baseDilations[i])
|
|
baseDilations = builder.EndVector()
|
|
if self.windowDilations is not None:
|
|
if np is not None and type(self.windowDilations) is np.ndarray:
|
|
windowDilations = builder.CreateNumpyVector(self.windowDilations)
|
|
else:
|
|
StablehloReduceWindowOptionsStartWindowDilationsVector(builder, len(self.windowDilations))
|
|
for i in reversed(range(len(self.windowDilations))):
|
|
builder.PrependInt64(self.windowDilations[i])
|
|
windowDilations = builder.EndVector()
|
|
if self.padding is not None:
|
|
if np is not None and type(self.padding) is np.ndarray:
|
|
padding = builder.CreateNumpyVector(self.padding)
|
|
else:
|
|
StablehloReduceWindowOptionsStartPaddingVector(builder, len(self.padding))
|
|
for i in reversed(range(len(self.padding))):
|
|
builder.PrependInt64(self.padding[i])
|
|
padding = builder.EndVector()
|
|
StablehloReduceWindowOptionsStart(builder)
|
|
if self.windowDimensions is not None:
|
|
StablehloReduceWindowOptionsAddWindowDimensions(builder, windowDimensions)
|
|
if self.windowStrides is not None:
|
|
StablehloReduceWindowOptionsAddWindowStrides(builder, windowStrides)
|
|
if self.baseDilations is not None:
|
|
StablehloReduceWindowOptionsAddBaseDilations(builder, baseDilations)
|
|
if self.windowDilations is not None:
|
|
StablehloReduceWindowOptionsAddWindowDilations(builder, windowDilations)
|
|
if self.padding is not None:
|
|
StablehloReduceWindowOptionsAddPadding(builder, padding)
|
|
StablehloReduceWindowOptionsAddBodySubgraphIndex(builder, self.bodySubgraphIndex)
|
|
stablehloReduceWindowOptions = StablehloReduceWindowOptionsEnd(builder)
|
|
return stablehloReduceWindowOptions
|
|
|
|
|
|
class StablehloWhileOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloWhileOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloWhileOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloWhileOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloWhileOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloWhileOptions
|
|
def CondSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloWhileOptions
|
|
def BodySubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloWhileOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def StablehloWhileOptionsAddCondSubgraphIndex(builder, condSubgraphIndex):
|
|
builder.PrependInt32Slot(0, condSubgraphIndex, 0)
|
|
|
|
def StablehloWhileOptionsAddBodySubgraphIndex(builder, bodySubgraphIndex):
|
|
builder.PrependInt32Slot(1, bodySubgraphIndex, 0)
|
|
|
|
def StablehloWhileOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloWhileOptionsT(object):
|
|
|
|
# StablehloWhileOptionsT
|
|
def __init__(self):
|
|
self.condSubgraphIndex = 0 # type: int
|
|
self.bodySubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloWhileOptions = StablehloWhileOptions()
|
|
stablehloWhileOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloWhileOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloWhileOptions):
|
|
x = StablehloWhileOptionsT()
|
|
x._UnPack(stablehloWhileOptions)
|
|
return x
|
|
|
|
# StablehloWhileOptionsT
|
|
def _UnPack(self, stablehloWhileOptions):
|
|
if stablehloWhileOptions is None:
|
|
return
|
|
self.condSubgraphIndex = stablehloWhileOptions.CondSubgraphIndex()
|
|
self.bodySubgraphIndex = stablehloWhileOptions.BodySubgraphIndex()
|
|
|
|
# StablehloWhileOptionsT
|
|
def Pack(self, builder):
|
|
StablehloWhileOptionsStart(builder)
|
|
StablehloWhileOptionsAddCondSubgraphIndex(builder, self.condSubgraphIndex)
|
|
StablehloWhileOptionsAddBodySubgraphIndex(builder, self.bodySubgraphIndex)
|
|
stablehloWhileOptions = StablehloWhileOptionsEnd(builder)
|
|
return stablehloWhileOptions
|
|
|
|
|
|
class StablehloSortOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloSortOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloSortOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloSortOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloSortOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloSortOptions
|
|
def Dimension(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloSortOptions
|
|
def IsStable(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# StablehloSortOptions
|
|
def ComparatorSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloSortOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def StablehloSortOptionsAddDimension(builder, dimension):
|
|
builder.PrependInt64Slot(0, dimension, 0)
|
|
|
|
def StablehloSortOptionsAddIsStable(builder, isStable):
|
|
builder.PrependBoolSlot(1, isStable, 0)
|
|
|
|
def StablehloSortOptionsAddComparatorSubgraphIndex(builder, comparatorSubgraphIndex):
|
|
builder.PrependInt32Slot(2, comparatorSubgraphIndex, 0)
|
|
|
|
def StablehloSortOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloSortOptionsT(object):
|
|
|
|
# StablehloSortOptionsT
|
|
def __init__(self):
|
|
self.dimension = 0 # type: int
|
|
self.isStable = False # type: bool
|
|
self.comparatorSubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloSortOptions = StablehloSortOptions()
|
|
stablehloSortOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloSortOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloSortOptions):
|
|
x = StablehloSortOptionsT()
|
|
x._UnPack(stablehloSortOptions)
|
|
return x
|
|
|
|
# StablehloSortOptionsT
|
|
def _UnPack(self, stablehloSortOptions):
|
|
if stablehloSortOptions is None:
|
|
return
|
|
self.dimension = stablehloSortOptions.Dimension()
|
|
self.isStable = stablehloSortOptions.IsStable()
|
|
self.comparatorSubgraphIndex = stablehloSortOptions.ComparatorSubgraphIndex()
|
|
|
|
# StablehloSortOptionsT
|
|
def Pack(self, builder):
|
|
StablehloSortOptionsStart(builder)
|
|
StablehloSortOptionsAddDimension(builder, self.dimension)
|
|
StablehloSortOptionsAddIsStable(builder, self.isStable)
|
|
StablehloSortOptionsAddComparatorSubgraphIndex(builder, self.comparatorSubgraphIndex)
|
|
stablehloSortOptions = StablehloSortOptionsEnd(builder)
|
|
return stablehloSortOptions
|
|
|
|
|
|
class StablehloConcatenateOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloConcatenateOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloConcatenateOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloConcatenateOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloConcatenateOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloConcatenateOptions
|
|
def Dimension(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloConcatenateOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloConcatenateOptionsAddDimension(builder, dimension):
|
|
builder.PrependInt64Slot(0, dimension, 0)
|
|
|
|
def StablehloConcatenateOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloConcatenateOptionsT(object):
|
|
|
|
# StablehloConcatenateOptionsT
|
|
def __init__(self):
|
|
self.dimension = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloConcatenateOptions = StablehloConcatenateOptions()
|
|
stablehloConcatenateOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloConcatenateOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloConcatenateOptions):
|
|
x = StablehloConcatenateOptionsT()
|
|
x._UnPack(stablehloConcatenateOptions)
|
|
return x
|
|
|
|
# StablehloConcatenateOptionsT
|
|
def _UnPack(self, stablehloConcatenateOptions):
|
|
if stablehloConcatenateOptions is None:
|
|
return
|
|
self.dimension = stablehloConcatenateOptions.Dimension()
|
|
|
|
# StablehloConcatenateOptionsT
|
|
def Pack(self, builder):
|
|
StablehloConcatenateOptionsStart(builder)
|
|
StablehloConcatenateOptionsAddDimension(builder, self.dimension)
|
|
stablehloConcatenateOptions = StablehloConcatenateOptionsEnd(builder)
|
|
return stablehloConcatenateOptions
|
|
|
|
|
|
class StablehloBroadcastInDimOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloBroadcastInDimOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloBroadcastInDimOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloBroadcastInDimOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloBroadcastInDimOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloBroadcastInDimOptions
|
|
def BroadcastDimensions(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
|
|
|
|
# StablehloBroadcastInDimOptions
|
|
def BroadcastDimensionsAsNumpy(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
|
|
|
|
# StablehloBroadcastInDimOptions
|
|
def BroadcastDimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloBroadcastInDimOptions
|
|
def BroadcastDimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def StablehloBroadcastInDimOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloBroadcastInDimOptionsAddBroadcastDimensions(builder, broadcastDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(broadcastDimensions), 0)
|
|
|
|
def StablehloBroadcastInDimOptionsStartBroadcastDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloBroadcastInDimOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloBroadcastInDimOptionsT(object):
|
|
|
|
# StablehloBroadcastInDimOptionsT
|
|
def __init__(self):
|
|
self.broadcastDimensions = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloBroadcastInDimOptions = StablehloBroadcastInDimOptions()
|
|
stablehloBroadcastInDimOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloBroadcastInDimOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloBroadcastInDimOptions):
|
|
x = StablehloBroadcastInDimOptionsT()
|
|
x._UnPack(stablehloBroadcastInDimOptions)
|
|
return x
|
|
|
|
# StablehloBroadcastInDimOptionsT
|
|
def _UnPack(self, stablehloBroadcastInDimOptions):
|
|
if stablehloBroadcastInDimOptions is None:
|
|
return
|
|
if not stablehloBroadcastInDimOptions.BroadcastDimensionsIsNone():
|
|
if np is None:
|
|
self.broadcastDimensions = []
|
|
for i in range(stablehloBroadcastInDimOptions.BroadcastDimensionsLength()):
|
|
self.broadcastDimensions.append(stablehloBroadcastInDimOptions.BroadcastDimensions(i))
|
|
else:
|
|
self.broadcastDimensions = stablehloBroadcastInDimOptions.BroadcastDimensionsAsNumpy()
|
|
|
|
# StablehloBroadcastInDimOptionsT
|
|
def Pack(self, builder):
|
|
if self.broadcastDimensions is not None:
|
|
if np is not None and type(self.broadcastDimensions) is np.ndarray:
|
|
broadcastDimensions = builder.CreateNumpyVector(self.broadcastDimensions)
|
|
else:
|
|
StablehloBroadcastInDimOptionsStartBroadcastDimensionsVector(builder, len(self.broadcastDimensions))
|
|
for i in reversed(range(len(self.broadcastDimensions))):
|
|
builder.PrependInt64(self.broadcastDimensions[i])
|
|
broadcastDimensions = builder.EndVector()
|
|
StablehloBroadcastInDimOptionsStart(builder)
|
|
if self.broadcastDimensions is not None:
|
|
StablehloBroadcastInDimOptionsAddBroadcastDimensions(builder, broadcastDimensions)
|
|
stablehloBroadcastInDimOptions = StablehloBroadcastInDimOptionsEnd(builder)
|
|
return stablehloBroadcastInDimOptions
|
|
|
|
|
|
class StablehloCompareOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloCompareOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloCompareOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloCompareOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloCompareOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloCompareOptions
|
|
def ComparisonDirection(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloCompareOptions
|
|
def CompareType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloCompareOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def StablehloCompareOptionsAddComparisonDirection(builder, comparisonDirection):
|
|
builder.PrependUint32Slot(0, comparisonDirection, 0)
|
|
|
|
def StablehloCompareOptionsAddCompareType(builder, compareType):
|
|
builder.PrependUint32Slot(1, compareType, 0)
|
|
|
|
def StablehloCompareOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloCompareOptionsT(object):
|
|
|
|
# StablehloCompareOptionsT
|
|
def __init__(self):
|
|
self.comparisonDirection = 0 # type: int
|
|
self.compareType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloCompareOptions = StablehloCompareOptions()
|
|
stablehloCompareOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloCompareOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloCompareOptions):
|
|
x = StablehloCompareOptionsT()
|
|
x._UnPack(stablehloCompareOptions)
|
|
return x
|
|
|
|
# StablehloCompareOptionsT
|
|
def _UnPack(self, stablehloCompareOptions):
|
|
if stablehloCompareOptions is None:
|
|
return
|
|
self.comparisonDirection = stablehloCompareOptions.ComparisonDirection()
|
|
self.compareType = stablehloCompareOptions.CompareType()
|
|
|
|
# StablehloCompareOptionsT
|
|
def Pack(self, builder):
|
|
StablehloCompareOptionsStart(builder)
|
|
StablehloCompareOptionsAddComparisonDirection(builder, self.comparisonDirection)
|
|
StablehloCompareOptionsAddCompareType(builder, self.compareType)
|
|
stablehloCompareOptions = StablehloCompareOptionsEnd(builder)
|
|
return stablehloCompareOptions
|
|
|
|
|
|
class StablehloDynamicSliceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloDynamicSliceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloDynamicSliceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloDynamicSliceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloDynamicSliceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloDynamicSliceOptions
|
|
def SliceSizes(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
|
|
|
|
# StablehloDynamicSliceOptions
|
|
def SliceSizesAsNumpy(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
|
|
|
|
# StablehloDynamicSliceOptions
|
|
def SliceSizesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloDynamicSliceOptions
|
|
def SliceSizesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def StablehloDynamicSliceOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloDynamicSliceOptionsAddSliceSizes(builder, sliceSizes):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(sliceSizes), 0)
|
|
|
|
def StablehloDynamicSliceOptionsStartSliceSizesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloDynamicSliceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloDynamicSliceOptionsT(object):
|
|
|
|
# StablehloDynamicSliceOptionsT
|
|
def __init__(self):
|
|
self.sliceSizes = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloDynamicSliceOptions = StablehloDynamicSliceOptions()
|
|
stablehloDynamicSliceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloDynamicSliceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloDynamicSliceOptions):
|
|
x = StablehloDynamicSliceOptionsT()
|
|
x._UnPack(stablehloDynamicSliceOptions)
|
|
return x
|
|
|
|
# StablehloDynamicSliceOptionsT
|
|
def _UnPack(self, stablehloDynamicSliceOptions):
|
|
if stablehloDynamicSliceOptions is None:
|
|
return
|
|
if not stablehloDynamicSliceOptions.SliceSizesIsNone():
|
|
if np is None:
|
|
self.sliceSizes = []
|
|
for i in range(stablehloDynamicSliceOptions.SliceSizesLength()):
|
|
self.sliceSizes.append(stablehloDynamicSliceOptions.SliceSizes(i))
|
|
else:
|
|
self.sliceSizes = stablehloDynamicSliceOptions.SliceSizesAsNumpy()
|
|
|
|
# StablehloDynamicSliceOptionsT
|
|
def Pack(self, builder):
|
|
if self.sliceSizes is not None:
|
|
if np is not None and type(self.sliceSizes) is np.ndarray:
|
|
sliceSizes = builder.CreateNumpyVector(self.sliceSizes)
|
|
else:
|
|
StablehloDynamicSliceOptionsStartSliceSizesVector(builder, len(self.sliceSizes))
|
|
for i in reversed(range(len(self.sliceSizes))):
|
|
builder.PrependInt64(self.sliceSizes[i])
|
|
sliceSizes = builder.EndVector()
|
|
StablehloDynamicSliceOptionsStart(builder)
|
|
if self.sliceSizes is not None:
|
|
StablehloDynamicSliceOptionsAddSliceSizes(builder, sliceSizes)
|
|
stablehloDynamicSliceOptions = StablehloDynamicSliceOptionsEnd(builder)
|
|
return stablehloDynamicSliceOptions
|
|
|
|
|
|
class StablehloPadOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloPadOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloPadOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloPadOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloPadOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingLow(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
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingLowAsNumpy(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
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingLowLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingLowIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingHigh(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
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingHighAsNumpy(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
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingHighLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloPadOptions
|
|
def EdgePaddingHighIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloPadOptions
|
|
def InteriorPadding(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
|
|
|
|
# StablehloPadOptions
|
|
def InteriorPaddingAsNumpy(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
|
|
|
|
# StablehloPadOptions
|
|
def InteriorPaddingLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloPadOptions
|
|
def InteriorPaddingIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
def StablehloPadOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def StablehloPadOptionsAddEdgePaddingLow(builder, edgePaddingLow):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(edgePaddingLow), 0)
|
|
|
|
def StablehloPadOptionsStartEdgePaddingLowVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloPadOptionsAddEdgePaddingHigh(builder, edgePaddingHigh):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(edgePaddingHigh), 0)
|
|
|
|
def StablehloPadOptionsStartEdgePaddingHighVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloPadOptionsAddInteriorPadding(builder, interiorPadding):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(interiorPadding), 0)
|
|
|
|
def StablehloPadOptionsStartInteriorPaddingVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloPadOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloPadOptionsT(object):
|
|
|
|
# StablehloPadOptionsT
|
|
def __init__(self):
|
|
self.edgePaddingLow = None # type: List[int]
|
|
self.edgePaddingHigh = None # type: List[int]
|
|
self.interiorPadding = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloPadOptions = StablehloPadOptions()
|
|
stablehloPadOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloPadOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloPadOptions):
|
|
x = StablehloPadOptionsT()
|
|
x._UnPack(stablehloPadOptions)
|
|
return x
|
|
|
|
# StablehloPadOptionsT
|
|
def _UnPack(self, stablehloPadOptions):
|
|
if stablehloPadOptions is None:
|
|
return
|
|
if not stablehloPadOptions.EdgePaddingLowIsNone():
|
|
if np is None:
|
|
self.edgePaddingLow = []
|
|
for i in range(stablehloPadOptions.EdgePaddingLowLength()):
|
|
self.edgePaddingLow.append(stablehloPadOptions.EdgePaddingLow(i))
|
|
else:
|
|
self.edgePaddingLow = stablehloPadOptions.EdgePaddingLowAsNumpy()
|
|
if not stablehloPadOptions.EdgePaddingHighIsNone():
|
|
if np is None:
|
|
self.edgePaddingHigh = []
|
|
for i in range(stablehloPadOptions.EdgePaddingHighLength()):
|
|
self.edgePaddingHigh.append(stablehloPadOptions.EdgePaddingHigh(i))
|
|
else:
|
|
self.edgePaddingHigh = stablehloPadOptions.EdgePaddingHighAsNumpy()
|
|
if not stablehloPadOptions.InteriorPaddingIsNone():
|
|
if np is None:
|
|
self.interiorPadding = []
|
|
for i in range(stablehloPadOptions.InteriorPaddingLength()):
|
|
self.interiorPadding.append(stablehloPadOptions.InteriorPadding(i))
|
|
else:
|
|
self.interiorPadding = stablehloPadOptions.InteriorPaddingAsNumpy()
|
|
|
|
# StablehloPadOptionsT
|
|
def Pack(self, builder):
|
|
if self.edgePaddingLow is not None:
|
|
if np is not None and type(self.edgePaddingLow) is np.ndarray:
|
|
edgePaddingLow = builder.CreateNumpyVector(self.edgePaddingLow)
|
|
else:
|
|
StablehloPadOptionsStartEdgePaddingLowVector(builder, len(self.edgePaddingLow))
|
|
for i in reversed(range(len(self.edgePaddingLow))):
|
|
builder.PrependInt64(self.edgePaddingLow[i])
|
|
edgePaddingLow = builder.EndVector()
|
|
if self.edgePaddingHigh is not None:
|
|
if np is not None and type(self.edgePaddingHigh) is np.ndarray:
|
|
edgePaddingHigh = builder.CreateNumpyVector(self.edgePaddingHigh)
|
|
else:
|
|
StablehloPadOptionsStartEdgePaddingHighVector(builder, len(self.edgePaddingHigh))
|
|
for i in reversed(range(len(self.edgePaddingHigh))):
|
|
builder.PrependInt64(self.edgePaddingHigh[i])
|
|
edgePaddingHigh = builder.EndVector()
|
|
if self.interiorPadding is not None:
|
|
if np is not None and type(self.interiorPadding) is np.ndarray:
|
|
interiorPadding = builder.CreateNumpyVector(self.interiorPadding)
|
|
else:
|
|
StablehloPadOptionsStartInteriorPaddingVector(builder, len(self.interiorPadding))
|
|
for i in reversed(range(len(self.interiorPadding))):
|
|
builder.PrependInt64(self.interiorPadding[i])
|
|
interiorPadding = builder.EndVector()
|
|
StablehloPadOptionsStart(builder)
|
|
if self.edgePaddingLow is not None:
|
|
StablehloPadOptionsAddEdgePaddingLow(builder, edgePaddingLow)
|
|
if self.edgePaddingHigh is not None:
|
|
StablehloPadOptionsAddEdgePaddingHigh(builder, edgePaddingHigh)
|
|
if self.interiorPadding is not None:
|
|
StablehloPadOptionsAddInteriorPadding(builder, interiorPadding)
|
|
stablehloPadOptions = StablehloPadOptionsEnd(builder)
|
|
return stablehloPadOptions
|
|
|
|
|
|
class StablehloIotaOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloIotaOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloIotaOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloIotaOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloIotaOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloIotaOptions
|
|
def IotaDimension(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloIotaOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloIotaOptionsAddIotaDimension(builder, iotaDimension):
|
|
builder.PrependInt64Slot(0, iotaDimension, 0)
|
|
|
|
def StablehloIotaOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloIotaOptionsT(object):
|
|
|
|
# StablehloIotaOptionsT
|
|
def __init__(self):
|
|
self.iotaDimension = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloIotaOptions = StablehloIotaOptions()
|
|
stablehloIotaOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloIotaOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloIotaOptions):
|
|
x = StablehloIotaOptionsT()
|
|
x._UnPack(stablehloIotaOptions)
|
|
return x
|
|
|
|
# StablehloIotaOptionsT
|
|
def _UnPack(self, stablehloIotaOptions):
|
|
if stablehloIotaOptions is None:
|
|
return
|
|
self.iotaDimension = stablehloIotaOptions.IotaDimension()
|
|
|
|
# StablehloIotaOptionsT
|
|
def Pack(self, builder):
|
|
StablehloIotaOptionsStart(builder)
|
|
StablehloIotaOptionsAddIotaDimension(builder, self.iotaDimension)
|
|
stablehloIotaOptions = StablehloIotaOptionsEnd(builder)
|
|
return stablehloIotaOptions
|
|
|
|
|
|
class StablehloCustomCallOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloCustomCallOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloCustomCallOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloCustomCallOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloCustomCallOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloCustomCallOptions
|
|
def CallTargetName(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# StablehloCustomCallOptions
|
|
def HasSideEffect(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# StablehloCustomCallOptions
|
|
def BackendConfig(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# StablehloCustomCallOptions
|
|
def ApiVersion(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CalledComputations(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CalledComputationsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CalledComputationsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CalledComputationsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
return o == 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CustomAttributes(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
a = self._tab.Vector(o)
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CustomAttributesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o)
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CustomAttributesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloCustomCallOptions
|
|
def CustomAttributesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
return o == 0
|
|
|
|
def StablehloCustomCallOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def StablehloCustomCallOptionsAddCallTargetName(builder, callTargetName):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(callTargetName), 0)
|
|
|
|
def StablehloCustomCallOptionsAddHasSideEffect(builder, hasSideEffect):
|
|
builder.PrependBoolSlot(1, hasSideEffect, 0)
|
|
|
|
def StablehloCustomCallOptionsAddBackendConfig(builder, backendConfig):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(backendConfig), 0)
|
|
|
|
def StablehloCustomCallOptionsAddApiVersion(builder, apiVersion):
|
|
builder.PrependInt32Slot(3, apiVersion, 0)
|
|
|
|
def StablehloCustomCallOptionsAddCalledComputations(builder, calledComputations):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(calledComputations), 0)
|
|
|
|
def StablehloCustomCallOptionsStartCalledComputationsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def StablehloCustomCallOptionsAddCustomAttributes(builder, customAttributes):
|
|
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(customAttributes), 0)
|
|
|
|
def StablehloCustomCallOptionsStartCustomAttributesVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def StablehloCustomCallOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloCustomCallOptionsT(object):
|
|
|
|
# StablehloCustomCallOptionsT
|
|
def __init__(self):
|
|
self.callTargetName = None # type: str
|
|
self.hasSideEffect = False # type: bool
|
|
self.backendConfig = None # type: str
|
|
self.apiVersion = 0 # type: int
|
|
self.calledComputations = None # type: List[int]
|
|
self.customAttributes = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloCustomCallOptions = StablehloCustomCallOptions()
|
|
stablehloCustomCallOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloCustomCallOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloCustomCallOptions):
|
|
x = StablehloCustomCallOptionsT()
|
|
x._UnPack(stablehloCustomCallOptions)
|
|
return x
|
|
|
|
# StablehloCustomCallOptionsT
|
|
def _UnPack(self, stablehloCustomCallOptions):
|
|
if stablehloCustomCallOptions is None:
|
|
return
|
|
self.callTargetName = stablehloCustomCallOptions.CallTargetName()
|
|
self.hasSideEffect = stablehloCustomCallOptions.HasSideEffect()
|
|
self.backendConfig = stablehloCustomCallOptions.BackendConfig()
|
|
self.apiVersion = stablehloCustomCallOptions.ApiVersion()
|
|
if not stablehloCustomCallOptions.CalledComputationsIsNone():
|
|
if np is None:
|
|
self.calledComputations = []
|
|
for i in range(stablehloCustomCallOptions.CalledComputationsLength()):
|
|
self.calledComputations.append(stablehloCustomCallOptions.CalledComputations(i))
|
|
else:
|
|
self.calledComputations = stablehloCustomCallOptions.CalledComputationsAsNumpy()
|
|
if not stablehloCustomCallOptions.CustomAttributesIsNone():
|
|
if np is None:
|
|
self.customAttributes = []
|
|
for i in range(stablehloCustomCallOptions.CustomAttributesLength()):
|
|
self.customAttributes.append(stablehloCustomCallOptions.CustomAttributes(i))
|
|
else:
|
|
self.customAttributes = stablehloCustomCallOptions.CustomAttributesAsNumpy()
|
|
|
|
# StablehloCustomCallOptionsT
|
|
def Pack(self, builder):
|
|
if self.callTargetName is not None:
|
|
callTargetName = builder.CreateString(self.callTargetName)
|
|
if self.backendConfig is not None:
|
|
backendConfig = builder.CreateString(self.backendConfig)
|
|
if self.calledComputations is not None:
|
|
if np is not None and type(self.calledComputations) is np.ndarray:
|
|
calledComputations = builder.CreateNumpyVector(self.calledComputations)
|
|
else:
|
|
StablehloCustomCallOptionsStartCalledComputationsVector(builder, len(self.calledComputations))
|
|
for i in reversed(range(len(self.calledComputations))):
|
|
builder.PrependInt32(self.calledComputations[i])
|
|
calledComputations = builder.EndVector()
|
|
if self.customAttributes is not None:
|
|
if np is not None and type(self.customAttributes) is np.ndarray:
|
|
customAttributes = builder.CreateNumpyVector(self.customAttributes)
|
|
else:
|
|
StablehloCustomCallOptionsStartCustomAttributesVector(builder, len(self.customAttributes))
|
|
for i in reversed(range(len(self.customAttributes))):
|
|
builder.PrependUint8(self.customAttributes[i])
|
|
customAttributes = builder.EndVector()
|
|
StablehloCustomCallOptionsStart(builder)
|
|
if self.callTargetName is not None:
|
|
StablehloCustomCallOptionsAddCallTargetName(builder, callTargetName)
|
|
StablehloCustomCallOptionsAddHasSideEffect(builder, self.hasSideEffect)
|
|
if self.backendConfig is not None:
|
|
StablehloCustomCallOptionsAddBackendConfig(builder, backendConfig)
|
|
StablehloCustomCallOptionsAddApiVersion(builder, self.apiVersion)
|
|
if self.calledComputations is not None:
|
|
StablehloCustomCallOptionsAddCalledComputations(builder, calledComputations)
|
|
if self.customAttributes is not None:
|
|
StablehloCustomCallOptionsAddCustomAttributes(builder, customAttributes)
|
|
stablehloCustomCallOptions = StablehloCustomCallOptionsEnd(builder)
|
|
return stablehloCustomCallOptions
|
|
|
|
|
|
class StablehloReduceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloReduceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloReduceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloReduceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloReduceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloReduceOptions
|
|
def Dimensions(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
|
|
|
|
# StablehloReduceOptions
|
|
def DimensionsAsNumpy(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
|
|
|
|
# StablehloReduceOptions
|
|
def DimensionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloReduceOptions
|
|
def DimensionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloReduceOptions
|
|
def BodySubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloReduceOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def StablehloReduceOptionsAddDimensions(builder, dimensions):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(dimensions), 0)
|
|
|
|
def StablehloReduceOptionsStartDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloReduceOptionsAddBodySubgraphIndex(builder, bodySubgraphIndex):
|
|
builder.PrependInt32Slot(1, bodySubgraphIndex, 0)
|
|
|
|
def StablehloReduceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloReduceOptionsT(object):
|
|
|
|
# StablehloReduceOptionsT
|
|
def __init__(self):
|
|
self.dimensions = None # type: List[int]
|
|
self.bodySubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloReduceOptions = StablehloReduceOptions()
|
|
stablehloReduceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloReduceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloReduceOptions):
|
|
x = StablehloReduceOptionsT()
|
|
x._UnPack(stablehloReduceOptions)
|
|
return x
|
|
|
|
# StablehloReduceOptionsT
|
|
def _UnPack(self, stablehloReduceOptions):
|
|
if stablehloReduceOptions is None:
|
|
return
|
|
if not stablehloReduceOptions.DimensionsIsNone():
|
|
if np is None:
|
|
self.dimensions = []
|
|
for i in range(stablehloReduceOptions.DimensionsLength()):
|
|
self.dimensions.append(stablehloReduceOptions.Dimensions(i))
|
|
else:
|
|
self.dimensions = stablehloReduceOptions.DimensionsAsNumpy()
|
|
self.bodySubgraphIndex = stablehloReduceOptions.BodySubgraphIndex()
|
|
|
|
# StablehloReduceOptionsT
|
|
def Pack(self, builder):
|
|
if self.dimensions is not None:
|
|
if np is not None and type(self.dimensions) is np.ndarray:
|
|
dimensions = builder.CreateNumpyVector(self.dimensions)
|
|
else:
|
|
StablehloReduceOptionsStartDimensionsVector(builder, len(self.dimensions))
|
|
for i in reversed(range(len(self.dimensions))):
|
|
builder.PrependInt64(self.dimensions[i])
|
|
dimensions = builder.EndVector()
|
|
StablehloReduceOptionsStart(builder)
|
|
if self.dimensions is not None:
|
|
StablehloReduceOptionsAddDimensions(builder, dimensions)
|
|
StablehloReduceOptionsAddBodySubgraphIndex(builder, self.bodySubgraphIndex)
|
|
stablehloReduceOptions = StablehloReduceOptionsEnd(builder)
|
|
return stablehloReduceOptions
|
|
|
|
|
|
class StablehloSliceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloSliceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloSliceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloSliceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloSliceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloSliceOptions
|
|
def StartIndices(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
|
|
|
|
# StablehloSliceOptions
|
|
def StartIndicesAsNumpy(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
|
|
|
|
# StablehloSliceOptions
|
|
def StartIndicesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloSliceOptions
|
|
def StartIndicesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# StablehloSliceOptions
|
|
def LimitIndices(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
|
|
|
|
# StablehloSliceOptions
|
|
def LimitIndicesAsNumpy(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
|
|
|
|
# StablehloSliceOptions
|
|
def LimitIndicesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloSliceOptions
|
|
def LimitIndicesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloSliceOptions
|
|
def Strides(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
|
|
|
|
# StablehloSliceOptions
|
|
def StridesAsNumpy(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
|
|
|
|
# StablehloSliceOptions
|
|
def StridesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloSliceOptions
|
|
def StridesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
def StablehloSliceOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def StablehloSliceOptionsAddStartIndices(builder, startIndices):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(startIndices), 0)
|
|
|
|
def StablehloSliceOptionsStartStartIndicesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloSliceOptionsAddLimitIndices(builder, limitIndices):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(limitIndices), 0)
|
|
|
|
def StablehloSliceOptionsStartLimitIndicesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloSliceOptionsAddStrides(builder, strides):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(strides), 0)
|
|
|
|
def StablehloSliceOptionsStartStridesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloSliceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloSliceOptionsT(object):
|
|
|
|
# StablehloSliceOptionsT
|
|
def __init__(self):
|
|
self.startIndices = None # type: List[int]
|
|
self.limitIndices = None # type: List[int]
|
|
self.strides = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloSliceOptions = StablehloSliceOptions()
|
|
stablehloSliceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloSliceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloSliceOptions):
|
|
x = StablehloSliceOptionsT()
|
|
x._UnPack(stablehloSliceOptions)
|
|
return x
|
|
|
|
# StablehloSliceOptionsT
|
|
def _UnPack(self, stablehloSliceOptions):
|
|
if stablehloSliceOptions is None:
|
|
return
|
|
if not stablehloSliceOptions.StartIndicesIsNone():
|
|
if np is None:
|
|
self.startIndices = []
|
|
for i in range(stablehloSliceOptions.StartIndicesLength()):
|
|
self.startIndices.append(stablehloSliceOptions.StartIndices(i))
|
|
else:
|
|
self.startIndices = stablehloSliceOptions.StartIndicesAsNumpy()
|
|
if not stablehloSliceOptions.LimitIndicesIsNone():
|
|
if np is None:
|
|
self.limitIndices = []
|
|
for i in range(stablehloSliceOptions.LimitIndicesLength()):
|
|
self.limitIndices.append(stablehloSliceOptions.LimitIndices(i))
|
|
else:
|
|
self.limitIndices = stablehloSliceOptions.LimitIndicesAsNumpy()
|
|
if not stablehloSliceOptions.StridesIsNone():
|
|
if np is None:
|
|
self.strides = []
|
|
for i in range(stablehloSliceOptions.StridesLength()):
|
|
self.strides.append(stablehloSliceOptions.Strides(i))
|
|
else:
|
|
self.strides = stablehloSliceOptions.StridesAsNumpy()
|
|
|
|
# StablehloSliceOptionsT
|
|
def Pack(self, builder):
|
|
if self.startIndices is not None:
|
|
if np is not None and type(self.startIndices) is np.ndarray:
|
|
startIndices = builder.CreateNumpyVector(self.startIndices)
|
|
else:
|
|
StablehloSliceOptionsStartStartIndicesVector(builder, len(self.startIndices))
|
|
for i in reversed(range(len(self.startIndices))):
|
|
builder.PrependInt64(self.startIndices[i])
|
|
startIndices = builder.EndVector()
|
|
if self.limitIndices is not None:
|
|
if np is not None and type(self.limitIndices) is np.ndarray:
|
|
limitIndices = builder.CreateNumpyVector(self.limitIndices)
|
|
else:
|
|
StablehloSliceOptionsStartLimitIndicesVector(builder, len(self.limitIndices))
|
|
for i in reversed(range(len(self.limitIndices))):
|
|
builder.PrependInt64(self.limitIndices[i])
|
|
limitIndices = builder.EndVector()
|
|
if self.strides is not None:
|
|
if np is not None and type(self.strides) is np.ndarray:
|
|
strides = builder.CreateNumpyVector(self.strides)
|
|
else:
|
|
StablehloSliceOptionsStartStridesVector(builder, len(self.strides))
|
|
for i in reversed(range(len(self.strides))):
|
|
builder.PrependInt64(self.strides[i])
|
|
strides = builder.EndVector()
|
|
StablehloSliceOptionsStart(builder)
|
|
if self.startIndices is not None:
|
|
StablehloSliceOptionsAddStartIndices(builder, startIndices)
|
|
if self.limitIndices is not None:
|
|
StablehloSliceOptionsAddLimitIndices(builder, limitIndices)
|
|
if self.strides is not None:
|
|
StablehloSliceOptionsAddStrides(builder, strides)
|
|
stablehloSliceOptions = StablehloSliceOptionsEnd(builder)
|
|
return stablehloSliceOptions
|
|
|
|
|
|
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 StablehloConvolutionOptionsAddWindowStrides(builder, windowStrides):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(windowStrides), 0)
|
|
|
|
def StablehloConvolutionOptionsStartWindowStridesVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddPadding(builder, padding):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(padding), 0)
|
|
|
|
def StablehloConvolutionOptionsStartPaddingVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddLhsDilation(builder, lhsDilation):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(lhsDilation), 0)
|
|
|
|
def StablehloConvolutionOptionsStartLhsDilationVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddRhsDilation(builder, rhsDilation):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(rhsDilation), 0)
|
|
|
|
def StablehloConvolutionOptionsStartRhsDilationVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddWindowReversal(builder, windowReversal):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(windowReversal), 0)
|
|
|
|
def StablehloConvolutionOptionsStartWindowReversalVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def StablehloConvolutionOptionsAddInputBatchDimension(builder, inputBatchDimension):
|
|
builder.PrependInt64Slot(5, inputBatchDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddInputFeatureDimension(builder, inputFeatureDimension):
|
|
builder.PrependInt64Slot(6, inputFeatureDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddInputSpatialDimensions(builder, inputSpatialDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(inputSpatialDimensions), 0)
|
|
|
|
def StablehloConvolutionOptionsStartInputSpatialDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddKernelInputFeatureDimension(builder, kernelInputFeatureDimension):
|
|
builder.PrependInt64Slot(8, kernelInputFeatureDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddKernelOutputFeatureDimension(builder, kernelOutputFeatureDimension):
|
|
builder.PrependInt64Slot(9, kernelOutputFeatureDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddKernelSpatialDimensions(builder, kernelSpatialDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(10, flatbuffers.number_types.UOffsetTFlags.py_type(kernelSpatialDimensions), 0)
|
|
|
|
def StablehloConvolutionOptionsStartKernelSpatialDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddOutputBatchDimension(builder, outputBatchDimension):
|
|
builder.PrependInt64Slot(11, outputBatchDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddOutputFeatureDimension(builder, outputFeatureDimension):
|
|
builder.PrependInt64Slot(12, outputFeatureDimension, 0)
|
|
|
|
def StablehloConvolutionOptionsAddOutputSpatialDimensions(builder, outputSpatialDimensions):
|
|
builder.PrependUOffsetTRelativeSlot(13, flatbuffers.number_types.UOffsetTFlags.py_type(outputSpatialDimensions), 0)
|
|
|
|
def StablehloConvolutionOptionsStartOutputSpatialDimensionsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloConvolutionOptionsAddFeatureGroupCount(builder, featureGroupCount):
|
|
builder.PrependInt64Slot(14, featureGroupCount, 0)
|
|
|
|
def StablehloConvolutionOptionsAddBatchGroupCount(builder, batchGroupCount):
|
|
builder.PrependInt64Slot(15, batchGroupCount, 0)
|
|
|
|
def StablehloConvolutionOptionsAddPrecisionConfig(builder, precisionConfig):
|
|
builder.PrependUOffsetTRelativeSlot(16, flatbuffers.number_types.UOffsetTFlags.py_type(precisionConfig), 0)
|
|
|
|
def StablehloConvolutionOptionsStartPrecisionConfigVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def StablehloConvolutionOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloConvolutionOptionsT(object):
|
|
|
|
# StablehloConvolutionOptionsT
|
|
def __init__(self):
|
|
self.windowStrides = None # type: List[int]
|
|
self.padding = None # type: List[int]
|
|
self.lhsDilation = None # type: List[int]
|
|
self.rhsDilation = None # type: List[int]
|
|
self.windowReversal = None # type: List[bool]
|
|
self.inputBatchDimension = 0 # type: int
|
|
self.inputFeatureDimension = 0 # type: int
|
|
self.inputSpatialDimensions = None # type: List[int]
|
|
self.kernelInputFeatureDimension = 0 # type: int
|
|
self.kernelOutputFeatureDimension = 0 # type: int
|
|
self.kernelSpatialDimensions = None # type: List[int]
|
|
self.outputBatchDimension = 0 # type: int
|
|
self.outputFeatureDimension = 0 # type: int
|
|
self.outputSpatialDimensions = None # type: List[int]
|
|
self.featureGroupCount = 0 # type: int
|
|
self.batchGroupCount = 0 # type: int
|
|
self.precisionConfig = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloConvolutionOptions = StablehloConvolutionOptions()
|
|
stablehloConvolutionOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloConvolutionOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloConvolutionOptions):
|
|
x = StablehloConvolutionOptionsT()
|
|
x._UnPack(stablehloConvolutionOptions)
|
|
return x
|
|
|
|
# StablehloConvolutionOptionsT
|
|
def _UnPack(self, stablehloConvolutionOptions):
|
|
if stablehloConvolutionOptions is None:
|
|
return
|
|
if not stablehloConvolutionOptions.WindowStridesIsNone():
|
|
if np is None:
|
|
self.windowStrides = []
|
|
for i in range(stablehloConvolutionOptions.WindowStridesLength()):
|
|
self.windowStrides.append(stablehloConvolutionOptions.WindowStrides(i))
|
|
else:
|
|
self.windowStrides = stablehloConvolutionOptions.WindowStridesAsNumpy()
|
|
if not stablehloConvolutionOptions.PaddingIsNone():
|
|
if np is None:
|
|
self.padding = []
|
|
for i in range(stablehloConvolutionOptions.PaddingLength()):
|
|
self.padding.append(stablehloConvolutionOptions.Padding(i))
|
|
else:
|
|
self.padding = stablehloConvolutionOptions.PaddingAsNumpy()
|
|
if not stablehloConvolutionOptions.LhsDilationIsNone():
|
|
if np is None:
|
|
self.lhsDilation = []
|
|
for i in range(stablehloConvolutionOptions.LhsDilationLength()):
|
|
self.lhsDilation.append(stablehloConvolutionOptions.LhsDilation(i))
|
|
else:
|
|
self.lhsDilation = stablehloConvolutionOptions.LhsDilationAsNumpy()
|
|
if not stablehloConvolutionOptions.RhsDilationIsNone():
|
|
if np is None:
|
|
self.rhsDilation = []
|
|
for i in range(stablehloConvolutionOptions.RhsDilationLength()):
|
|
self.rhsDilation.append(stablehloConvolutionOptions.RhsDilation(i))
|
|
else:
|
|
self.rhsDilation = stablehloConvolutionOptions.RhsDilationAsNumpy()
|
|
if not stablehloConvolutionOptions.WindowReversalIsNone():
|
|
if np is None:
|
|
self.windowReversal = []
|
|
for i in range(stablehloConvolutionOptions.WindowReversalLength()):
|
|
self.windowReversal.append(stablehloConvolutionOptions.WindowReversal(i))
|
|
else:
|
|
self.windowReversal = stablehloConvolutionOptions.WindowReversalAsNumpy()
|
|
self.inputBatchDimension = stablehloConvolutionOptions.InputBatchDimension()
|
|
self.inputFeatureDimension = stablehloConvolutionOptions.InputFeatureDimension()
|
|
if not stablehloConvolutionOptions.InputSpatialDimensionsIsNone():
|
|
if np is None:
|
|
self.inputSpatialDimensions = []
|
|
for i in range(stablehloConvolutionOptions.InputSpatialDimensionsLength()):
|
|
self.inputSpatialDimensions.append(stablehloConvolutionOptions.InputSpatialDimensions(i))
|
|
else:
|
|
self.inputSpatialDimensions = stablehloConvolutionOptions.InputSpatialDimensionsAsNumpy()
|
|
self.kernelInputFeatureDimension = stablehloConvolutionOptions.KernelInputFeatureDimension()
|
|
self.kernelOutputFeatureDimension = stablehloConvolutionOptions.KernelOutputFeatureDimension()
|
|
if not stablehloConvolutionOptions.KernelSpatialDimensionsIsNone():
|
|
if np is None:
|
|
self.kernelSpatialDimensions = []
|
|
for i in range(stablehloConvolutionOptions.KernelSpatialDimensionsLength()):
|
|
self.kernelSpatialDimensions.append(stablehloConvolutionOptions.KernelSpatialDimensions(i))
|
|
else:
|
|
self.kernelSpatialDimensions = stablehloConvolutionOptions.KernelSpatialDimensionsAsNumpy()
|
|
self.outputBatchDimension = stablehloConvolutionOptions.OutputBatchDimension()
|
|
self.outputFeatureDimension = stablehloConvolutionOptions.OutputFeatureDimension()
|
|
if not stablehloConvolutionOptions.OutputSpatialDimensionsIsNone():
|
|
if np is None:
|
|
self.outputSpatialDimensions = []
|
|
for i in range(stablehloConvolutionOptions.OutputSpatialDimensionsLength()):
|
|
self.outputSpatialDimensions.append(stablehloConvolutionOptions.OutputSpatialDimensions(i))
|
|
else:
|
|
self.outputSpatialDimensions = stablehloConvolutionOptions.OutputSpatialDimensionsAsNumpy()
|
|
self.featureGroupCount = stablehloConvolutionOptions.FeatureGroupCount()
|
|
self.batchGroupCount = stablehloConvolutionOptions.BatchGroupCount()
|
|
if not stablehloConvolutionOptions.PrecisionConfigIsNone():
|
|
if np is None:
|
|
self.precisionConfig = []
|
|
for i in range(stablehloConvolutionOptions.PrecisionConfigLength()):
|
|
self.precisionConfig.append(stablehloConvolutionOptions.PrecisionConfig(i))
|
|
else:
|
|
self.precisionConfig = stablehloConvolutionOptions.PrecisionConfigAsNumpy()
|
|
|
|
# StablehloConvolutionOptionsT
|
|
def Pack(self, builder):
|
|
if self.windowStrides is not None:
|
|
if np is not None and type(self.windowStrides) is np.ndarray:
|
|
windowStrides = builder.CreateNumpyVector(self.windowStrides)
|
|
else:
|
|
StablehloConvolutionOptionsStartWindowStridesVector(builder, len(self.windowStrides))
|
|
for i in reversed(range(len(self.windowStrides))):
|
|
builder.PrependInt64(self.windowStrides[i])
|
|
windowStrides = builder.EndVector()
|
|
if self.padding is not None:
|
|
if np is not None and type(self.padding) is np.ndarray:
|
|
padding = builder.CreateNumpyVector(self.padding)
|
|
else:
|
|
StablehloConvolutionOptionsStartPaddingVector(builder, len(self.padding))
|
|
for i in reversed(range(len(self.padding))):
|
|
builder.PrependInt64(self.padding[i])
|
|
padding = builder.EndVector()
|
|
if self.lhsDilation is not None:
|
|
if np is not None and type(self.lhsDilation) is np.ndarray:
|
|
lhsDilation = builder.CreateNumpyVector(self.lhsDilation)
|
|
else:
|
|
StablehloConvolutionOptionsStartLhsDilationVector(builder, len(self.lhsDilation))
|
|
for i in reversed(range(len(self.lhsDilation))):
|
|
builder.PrependInt64(self.lhsDilation[i])
|
|
lhsDilation = builder.EndVector()
|
|
if self.rhsDilation is not None:
|
|
if np is not None and type(self.rhsDilation) is np.ndarray:
|
|
rhsDilation = builder.CreateNumpyVector(self.rhsDilation)
|
|
else:
|
|
StablehloConvolutionOptionsStartRhsDilationVector(builder, len(self.rhsDilation))
|
|
for i in reversed(range(len(self.rhsDilation))):
|
|
builder.PrependInt64(self.rhsDilation[i])
|
|
rhsDilation = builder.EndVector()
|
|
if self.windowReversal is not None:
|
|
if np is not None and type(self.windowReversal) is np.ndarray:
|
|
windowReversal = builder.CreateNumpyVector(self.windowReversal)
|
|
else:
|
|
StablehloConvolutionOptionsStartWindowReversalVector(builder, len(self.windowReversal))
|
|
for i in reversed(range(len(self.windowReversal))):
|
|
builder.PrependBool(self.windowReversal[i])
|
|
windowReversal = builder.EndVector()
|
|
if self.inputSpatialDimensions is not None:
|
|
if np is not None and type(self.inputSpatialDimensions) is np.ndarray:
|
|
inputSpatialDimensions = builder.CreateNumpyVector(self.inputSpatialDimensions)
|
|
else:
|
|
StablehloConvolutionOptionsStartInputSpatialDimensionsVector(builder, len(self.inputSpatialDimensions))
|
|
for i in reversed(range(len(self.inputSpatialDimensions))):
|
|
builder.PrependInt64(self.inputSpatialDimensions[i])
|
|
inputSpatialDimensions = builder.EndVector()
|
|
if self.kernelSpatialDimensions is not None:
|
|
if np is not None and type(self.kernelSpatialDimensions) is np.ndarray:
|
|
kernelSpatialDimensions = builder.CreateNumpyVector(self.kernelSpatialDimensions)
|
|
else:
|
|
StablehloConvolutionOptionsStartKernelSpatialDimensionsVector(builder, len(self.kernelSpatialDimensions))
|
|
for i in reversed(range(len(self.kernelSpatialDimensions))):
|
|
builder.PrependInt64(self.kernelSpatialDimensions[i])
|
|
kernelSpatialDimensions = builder.EndVector()
|
|
if self.outputSpatialDimensions is not None:
|
|
if np is not None and type(self.outputSpatialDimensions) is np.ndarray:
|
|
outputSpatialDimensions = builder.CreateNumpyVector(self.outputSpatialDimensions)
|
|
else:
|
|
StablehloConvolutionOptionsStartOutputSpatialDimensionsVector(builder, len(self.outputSpatialDimensions))
|
|
for i in reversed(range(len(self.outputSpatialDimensions))):
|
|
builder.PrependInt64(self.outputSpatialDimensions[i])
|
|
outputSpatialDimensions = builder.EndVector()
|
|
if self.precisionConfig is not None:
|
|
if np is not None and type(self.precisionConfig) is np.ndarray:
|
|
precisionConfig = builder.CreateNumpyVector(self.precisionConfig)
|
|
else:
|
|
StablehloConvolutionOptionsStartPrecisionConfigVector(builder, len(self.precisionConfig))
|
|
for i in reversed(range(len(self.precisionConfig))):
|
|
builder.PrependUint32(self.precisionConfig[i])
|
|
precisionConfig = builder.EndVector()
|
|
StablehloConvolutionOptionsStart(builder)
|
|
if self.windowStrides is not None:
|
|
StablehloConvolutionOptionsAddWindowStrides(builder, windowStrides)
|
|
if self.padding is not None:
|
|
StablehloConvolutionOptionsAddPadding(builder, padding)
|
|
if self.lhsDilation is not None:
|
|
StablehloConvolutionOptionsAddLhsDilation(builder, lhsDilation)
|
|
if self.rhsDilation is not None:
|
|
StablehloConvolutionOptionsAddRhsDilation(builder, rhsDilation)
|
|
if self.windowReversal is not None:
|
|
StablehloConvolutionOptionsAddWindowReversal(builder, windowReversal)
|
|
StablehloConvolutionOptionsAddInputBatchDimension(builder, self.inputBatchDimension)
|
|
StablehloConvolutionOptionsAddInputFeatureDimension(builder, self.inputFeatureDimension)
|
|
if self.inputSpatialDimensions is not None:
|
|
StablehloConvolutionOptionsAddInputSpatialDimensions(builder, inputSpatialDimensions)
|
|
StablehloConvolutionOptionsAddKernelInputFeatureDimension(builder, self.kernelInputFeatureDimension)
|
|
StablehloConvolutionOptionsAddKernelOutputFeatureDimension(builder, self.kernelOutputFeatureDimension)
|
|
if self.kernelSpatialDimensions is not None:
|
|
StablehloConvolutionOptionsAddKernelSpatialDimensions(builder, kernelSpatialDimensions)
|
|
StablehloConvolutionOptionsAddOutputBatchDimension(builder, self.outputBatchDimension)
|
|
StablehloConvolutionOptionsAddOutputFeatureDimension(builder, self.outputFeatureDimension)
|
|
if self.outputSpatialDimensions is not None:
|
|
StablehloConvolutionOptionsAddOutputSpatialDimensions(builder, outputSpatialDimensions)
|
|
StablehloConvolutionOptionsAddFeatureGroupCount(builder, self.featureGroupCount)
|
|
StablehloConvolutionOptionsAddBatchGroupCount(builder, self.batchGroupCount)
|
|
if self.precisionConfig is not None:
|
|
StablehloConvolutionOptionsAddPrecisionConfig(builder, precisionConfig)
|
|
stablehloConvolutionOptions = StablehloConvolutionOptionsEnd(builder)
|
|
return stablehloConvolutionOptions
|
|
|
|
|
|
class StablehloScatterOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloScatterOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloScatterOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloScatterOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloScatterOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloScatterOptions
|
|
def IndicesAreSorted(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# StablehloScatterOptions
|
|
def UpdateWindowDims(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
|
|
|
|
# StablehloScatterOptions
|
|
def UpdateWindowDimsAsNumpy(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
|
|
|
|
# StablehloScatterOptions
|
|
def UpdateWindowDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloScatterOptions
|
|
def UpdateWindowDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# StablehloScatterOptions
|
|
def InsertedWindowDims(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
|
|
|
|
# StablehloScatterOptions
|
|
def InsertedWindowDimsAsNumpy(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
|
|
|
|
# StablehloScatterOptions
|
|
def InsertedWindowDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloScatterOptions
|
|
def InsertedWindowDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# StablehloScatterOptions
|
|
def ScatterDimsToOperandDims(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
|
|
|
|
# StablehloScatterOptions
|
|
def ScatterDimsToOperandDimsAsNumpy(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
|
|
|
|
# StablehloScatterOptions
|
|
def ScatterDimsToOperandDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# StablehloScatterOptions
|
|
def ScatterDimsToOperandDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
return o == 0
|
|
|
|
# StablehloScatterOptions
|
|
def IndexVectorDim(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StablehloScatterOptions
|
|
def UniqueIndices(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# StablehloScatterOptions
|
|
def UpdateComputationSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloScatterOptionsStart(builder):
|
|
builder.StartObject(7)
|
|
|
|
def StablehloScatterOptionsAddIndicesAreSorted(builder, indicesAreSorted):
|
|
builder.PrependBoolSlot(0, indicesAreSorted, 0)
|
|
|
|
def StablehloScatterOptionsAddUpdateWindowDims(builder, updateWindowDims):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(updateWindowDims), 0)
|
|
|
|
def StablehloScatterOptionsStartUpdateWindowDimsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloScatterOptionsAddInsertedWindowDims(builder, insertedWindowDims):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(insertedWindowDims), 0)
|
|
|
|
def StablehloScatterOptionsStartInsertedWindowDimsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloScatterOptionsAddScatterDimsToOperandDims(builder, scatterDimsToOperandDims):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(scatterDimsToOperandDims), 0)
|
|
|
|
def StablehloScatterOptionsStartScatterDimsToOperandDimsVector(builder, numElems):
|
|
return builder.StartVector(8, numElems, 8)
|
|
|
|
def StablehloScatterOptionsAddIndexVectorDim(builder, indexVectorDim):
|
|
builder.PrependInt64Slot(4, indexVectorDim, 0)
|
|
|
|
def StablehloScatterOptionsAddUniqueIndices(builder, uniqueIndices):
|
|
builder.PrependBoolSlot(5, uniqueIndices, 0)
|
|
|
|
def StablehloScatterOptionsAddUpdateComputationSubgraphIndex(builder, updateComputationSubgraphIndex):
|
|
builder.PrependInt32Slot(6, updateComputationSubgraphIndex, 0)
|
|
|
|
def StablehloScatterOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class StablehloScatterOptionsT(object):
|
|
|
|
# StablehloScatterOptionsT
|
|
def __init__(self):
|
|
self.indicesAreSorted = False # type: bool
|
|
self.updateWindowDims = None # type: List[int]
|
|
self.insertedWindowDims = None # type: List[int]
|
|
self.scatterDimsToOperandDims = None # type: List[int]
|
|
self.indexVectorDim = 0 # type: int
|
|
self.uniqueIndices = False # type: bool
|
|
self.updateComputationSubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloScatterOptions = StablehloScatterOptions()
|
|
stablehloScatterOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloScatterOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloScatterOptions):
|
|
x = StablehloScatterOptionsT()
|
|
x._UnPack(stablehloScatterOptions)
|
|
return x
|
|
|
|
# StablehloScatterOptionsT
|
|
def _UnPack(self, stablehloScatterOptions):
|
|
if stablehloScatterOptions is None:
|
|
return
|
|
self.indicesAreSorted = stablehloScatterOptions.IndicesAreSorted()
|
|
if not stablehloScatterOptions.UpdateWindowDimsIsNone():
|
|
if np is None:
|
|
self.updateWindowDims = []
|
|
for i in range(stablehloScatterOptions.UpdateWindowDimsLength()):
|
|
self.updateWindowDims.append(stablehloScatterOptions.UpdateWindowDims(i))
|
|
else:
|
|
self.updateWindowDims = stablehloScatterOptions.UpdateWindowDimsAsNumpy()
|
|
if not stablehloScatterOptions.InsertedWindowDimsIsNone():
|
|
if np is None:
|
|
self.insertedWindowDims = []
|
|
for i in range(stablehloScatterOptions.InsertedWindowDimsLength()):
|
|
self.insertedWindowDims.append(stablehloScatterOptions.InsertedWindowDims(i))
|
|
else:
|
|
self.insertedWindowDims = stablehloScatterOptions.InsertedWindowDimsAsNumpy()
|
|
if not stablehloScatterOptions.ScatterDimsToOperandDimsIsNone():
|
|
if np is None:
|
|
self.scatterDimsToOperandDims = []
|
|
for i in range(stablehloScatterOptions.ScatterDimsToOperandDimsLength()):
|
|
self.scatterDimsToOperandDims.append(stablehloScatterOptions.ScatterDimsToOperandDims(i))
|
|
else:
|
|
self.scatterDimsToOperandDims = stablehloScatterOptions.ScatterDimsToOperandDimsAsNumpy()
|
|
self.indexVectorDim = stablehloScatterOptions.IndexVectorDim()
|
|
self.uniqueIndices = stablehloScatterOptions.UniqueIndices()
|
|
self.updateComputationSubgraphIndex = stablehloScatterOptions.UpdateComputationSubgraphIndex()
|
|
|
|
# StablehloScatterOptionsT
|
|
def Pack(self, builder):
|
|
if self.updateWindowDims is not None:
|
|
if np is not None and type(self.updateWindowDims) is np.ndarray:
|
|
updateWindowDims = builder.CreateNumpyVector(self.updateWindowDims)
|
|
else:
|
|
StablehloScatterOptionsStartUpdateWindowDimsVector(builder, len(self.updateWindowDims))
|
|
for i in reversed(range(len(self.updateWindowDims))):
|
|
builder.PrependInt64(self.updateWindowDims[i])
|
|
updateWindowDims = builder.EndVector()
|
|
if self.insertedWindowDims is not None:
|
|
if np is not None and type(self.insertedWindowDims) is np.ndarray:
|
|
insertedWindowDims = builder.CreateNumpyVector(self.insertedWindowDims)
|
|
else:
|
|
StablehloScatterOptionsStartInsertedWindowDimsVector(builder, len(self.insertedWindowDims))
|
|
for i in reversed(range(len(self.insertedWindowDims))):
|
|
builder.PrependInt64(self.insertedWindowDims[i])
|
|
insertedWindowDims = builder.EndVector()
|
|
if self.scatterDimsToOperandDims is not None:
|
|
if np is not None and type(self.scatterDimsToOperandDims) is np.ndarray:
|
|
scatterDimsToOperandDims = builder.CreateNumpyVector(self.scatterDimsToOperandDims)
|
|
else:
|
|
StablehloScatterOptionsStartScatterDimsToOperandDimsVector(builder, len(self.scatterDimsToOperandDims))
|
|
for i in reversed(range(len(self.scatterDimsToOperandDims))):
|
|
builder.PrependInt64(self.scatterDimsToOperandDims[i])
|
|
scatterDimsToOperandDims = builder.EndVector()
|
|
StablehloScatterOptionsStart(builder)
|
|
StablehloScatterOptionsAddIndicesAreSorted(builder, self.indicesAreSorted)
|
|
if self.updateWindowDims is not None:
|
|
StablehloScatterOptionsAddUpdateWindowDims(builder, updateWindowDims)
|
|
if self.insertedWindowDims is not None:
|
|
StablehloScatterOptionsAddInsertedWindowDims(builder, insertedWindowDims)
|
|
if self.scatterDimsToOperandDims is not None:
|
|
StablehloScatterOptionsAddScatterDimsToOperandDims(builder, scatterDimsToOperandDims)
|
|
StablehloScatterOptionsAddIndexVectorDim(builder, self.indexVectorDim)
|
|
StablehloScatterOptionsAddUniqueIndices(builder, self.uniqueIndices)
|
|
StablehloScatterOptionsAddUpdateComputationSubgraphIndex(builder, self.updateComputationSubgraphIndex)
|
|
stablehloScatterOptions = StablehloScatterOptionsEnd(builder)
|
|
return stablehloScatterOptions
|
|
|
|
|
|
class StablehloRngBitGeneratorOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StablehloRngBitGeneratorOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStablehloRngBitGeneratorOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StablehloRngBitGeneratorOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StablehloRngBitGeneratorOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StablehloRngBitGeneratorOptions
|
|
def Algorithm(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def StablehloRngBitGeneratorOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def StablehloRngBitGeneratorOptionsAddAlgorithm(builder, algorithm):
|
|
builder.PrependInt8Slot(0, algorithm, 0)
|
|
|
|
def StablehloRngBitGeneratorOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StablehloRngBitGeneratorOptionsT(object):
|
|
|
|
# StablehloRngBitGeneratorOptionsT
|
|
def __init__(self):
|
|
self.algorithm = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stablehloRngBitGeneratorOptions = StablehloRngBitGeneratorOptions()
|
|
stablehloRngBitGeneratorOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stablehloRngBitGeneratorOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stablehloRngBitGeneratorOptions):
|
|
x = StablehloRngBitGeneratorOptionsT()
|
|
x._UnPack(stablehloRngBitGeneratorOptions)
|
|
return x
|
|
|
|
# StablehloRngBitGeneratorOptionsT
|
|
def _UnPack(self, stablehloRngBitGeneratorOptions):
|
|
if stablehloRngBitGeneratorOptions is None:
|
|
return
|
|
self.algorithm = stablehloRngBitGeneratorOptions.Algorithm()
|
|
|
|
# StablehloRngBitGeneratorOptionsT
|
|
def Pack(self, builder):
|
|
StablehloRngBitGeneratorOptionsStart(builder)
|
|
StablehloRngBitGeneratorOptionsAddAlgorithm(builder, self.algorithm)
|
|
stablehloRngBitGeneratorOptions = StablehloRngBitGeneratorOptionsEnd(builder)
|
|
return stablehloRngBitGeneratorOptions
|
|
|
|
|
|
class Conv2DOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
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)
|
|
|
|
# Conv2DOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Conv2DOptions
|
|
def Padding(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv2DOptions
|
|
def StrideW(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv2DOptions
|
|
def StrideH(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv2DOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv2DOptions
|
|
def DilationWFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# Conv2DOptions
|
|
def DilationHFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# Conv2DOptions
|
|
def QuantizedBiasType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def Conv2DOptionsStart(builder):
|
|
builder.StartObject(7)
|
|
|
|
def Conv2DOptionsAddPadding(builder, padding):
|
|
builder.PrependInt8Slot(0, padding, 0)
|
|
|
|
def Conv2DOptionsAddStrideW(builder, strideW):
|
|
builder.PrependInt32Slot(1, strideW, 0)
|
|
|
|
def Conv2DOptionsAddStrideH(builder, strideH):
|
|
builder.PrependInt32Slot(2, strideH, 0)
|
|
|
|
def Conv2DOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(3, fusedActivationFunction, 0)
|
|
|
|
def Conv2DOptionsAddDilationWFactor(builder, dilationWFactor):
|
|
builder.PrependInt32Slot(4, dilationWFactor, 1)
|
|
|
|
def Conv2DOptionsAddDilationHFactor(builder, dilationHFactor):
|
|
builder.PrependInt32Slot(5, dilationHFactor, 1)
|
|
|
|
def Conv2DOptionsAddQuantizedBiasType(builder, quantizedBiasType):
|
|
builder.PrependInt8Slot(6, quantizedBiasType, 0)
|
|
|
|
def Conv2DOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class Conv2DOptionsT(object):
|
|
|
|
# Conv2DOptionsT
|
|
def __init__(self):
|
|
self.padding = 0 # type: int
|
|
self.strideW = 0 # type: int
|
|
self.strideH = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.dilationWFactor = 1 # type: int
|
|
self.dilationHFactor = 1 # type: int
|
|
self.quantizedBiasType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
conv2Doptions = Conv2DOptions()
|
|
conv2Doptions.Init(buf, pos)
|
|
return cls.InitFromObj(conv2Doptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, conv2Doptions):
|
|
x = Conv2DOptionsT()
|
|
x._UnPack(conv2Doptions)
|
|
return x
|
|
|
|
# Conv2DOptionsT
|
|
def _UnPack(self, conv2Doptions):
|
|
if conv2Doptions is None:
|
|
return
|
|
self.padding = conv2Doptions.Padding()
|
|
self.strideW = conv2Doptions.StrideW()
|
|
self.strideH = conv2Doptions.StrideH()
|
|
self.fusedActivationFunction = conv2Doptions.FusedActivationFunction()
|
|
self.dilationWFactor = conv2Doptions.DilationWFactor()
|
|
self.dilationHFactor = conv2Doptions.DilationHFactor()
|
|
self.quantizedBiasType = conv2Doptions.QuantizedBiasType()
|
|
|
|
# Conv2DOptionsT
|
|
def Pack(self, builder):
|
|
Conv2DOptionsStart(builder)
|
|
Conv2DOptionsAddPadding(builder, self.padding)
|
|
Conv2DOptionsAddStrideW(builder, self.strideW)
|
|
Conv2DOptionsAddStrideH(builder, self.strideH)
|
|
Conv2DOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
Conv2DOptionsAddDilationWFactor(builder, self.dilationWFactor)
|
|
Conv2DOptionsAddDilationHFactor(builder, self.dilationHFactor)
|
|
Conv2DOptionsAddQuantizedBiasType(builder, self.quantizedBiasType)
|
|
conv2Doptions = Conv2DOptionsEnd(builder)
|
|
return conv2Doptions
|
|
|
|
|
|
class Conv3DOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Conv3DOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsConv3DOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Conv3DOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Conv3DOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Conv3DOptions
|
|
def Padding(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv3DOptions
|
|
def StrideD(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv3DOptions
|
|
def StrideW(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv3DOptions
|
|
def StrideH(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv3DOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Conv3DOptions
|
|
def DilationDFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# Conv3DOptions
|
|
def DilationWFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# Conv3DOptions
|
|
def DilationHFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
def Conv3DOptionsStart(builder):
|
|
builder.StartObject(8)
|
|
|
|
def Conv3DOptionsAddPadding(builder, padding):
|
|
builder.PrependInt8Slot(0, padding, 0)
|
|
|
|
def Conv3DOptionsAddStrideD(builder, strideD):
|
|
builder.PrependInt32Slot(1, strideD, 0)
|
|
|
|
def Conv3DOptionsAddStrideW(builder, strideW):
|
|
builder.PrependInt32Slot(2, strideW, 0)
|
|
|
|
def Conv3DOptionsAddStrideH(builder, strideH):
|
|
builder.PrependInt32Slot(3, strideH, 0)
|
|
|
|
def Conv3DOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(4, fusedActivationFunction, 0)
|
|
|
|
def Conv3DOptionsAddDilationDFactor(builder, dilationDFactor):
|
|
builder.PrependInt32Slot(5, dilationDFactor, 1)
|
|
|
|
def Conv3DOptionsAddDilationWFactor(builder, dilationWFactor):
|
|
builder.PrependInt32Slot(6, dilationWFactor, 1)
|
|
|
|
def Conv3DOptionsAddDilationHFactor(builder, dilationHFactor):
|
|
builder.PrependInt32Slot(7, dilationHFactor, 1)
|
|
|
|
def Conv3DOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class Conv3DOptionsT(object):
|
|
|
|
# Conv3DOptionsT
|
|
def __init__(self):
|
|
self.padding = 0 # type: int
|
|
self.strideD = 0 # type: int
|
|
self.strideW = 0 # type: int
|
|
self.strideH = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.dilationDFactor = 1 # type: int
|
|
self.dilationWFactor = 1 # type: int
|
|
self.dilationHFactor = 1 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
conv3Doptions = Conv3DOptions()
|
|
conv3Doptions.Init(buf, pos)
|
|
return cls.InitFromObj(conv3Doptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, conv3Doptions):
|
|
x = Conv3DOptionsT()
|
|
x._UnPack(conv3Doptions)
|
|
return x
|
|
|
|
# Conv3DOptionsT
|
|
def _UnPack(self, conv3Doptions):
|
|
if conv3Doptions is None:
|
|
return
|
|
self.padding = conv3Doptions.Padding()
|
|
self.strideD = conv3Doptions.StrideD()
|
|
self.strideW = conv3Doptions.StrideW()
|
|
self.strideH = conv3Doptions.StrideH()
|
|
self.fusedActivationFunction = conv3Doptions.FusedActivationFunction()
|
|
self.dilationDFactor = conv3Doptions.DilationDFactor()
|
|
self.dilationWFactor = conv3Doptions.DilationWFactor()
|
|
self.dilationHFactor = conv3Doptions.DilationHFactor()
|
|
|
|
# Conv3DOptionsT
|
|
def Pack(self, builder):
|
|
Conv3DOptionsStart(builder)
|
|
Conv3DOptionsAddPadding(builder, self.padding)
|
|
Conv3DOptionsAddStrideD(builder, self.strideD)
|
|
Conv3DOptionsAddStrideW(builder, self.strideW)
|
|
Conv3DOptionsAddStrideH(builder, self.strideH)
|
|
Conv3DOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
Conv3DOptionsAddDilationDFactor(builder, self.dilationDFactor)
|
|
Conv3DOptionsAddDilationWFactor(builder, self.dilationWFactor)
|
|
Conv3DOptionsAddDilationHFactor(builder, self.dilationHFactor)
|
|
conv3Doptions = Conv3DOptionsEnd(builder)
|
|
return conv3Doptions
|
|
|
|
|
|
class Pool2DOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Pool2DOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsPool2DOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Pool2DOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Pool2DOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Pool2DOptions
|
|
def Padding(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Pool2DOptions
|
|
def StrideW(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Pool2DOptions
|
|
def StrideH(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Pool2DOptions
|
|
def FilterWidth(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Pool2DOptions
|
|
def FilterHeight(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Pool2DOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def Pool2DOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def Pool2DOptionsAddPadding(builder, padding):
|
|
builder.PrependInt8Slot(0, padding, 0)
|
|
|
|
def Pool2DOptionsAddStrideW(builder, strideW):
|
|
builder.PrependInt32Slot(1, strideW, 0)
|
|
|
|
def Pool2DOptionsAddStrideH(builder, strideH):
|
|
builder.PrependInt32Slot(2, strideH, 0)
|
|
|
|
def Pool2DOptionsAddFilterWidth(builder, filterWidth):
|
|
builder.PrependInt32Slot(3, filterWidth, 0)
|
|
|
|
def Pool2DOptionsAddFilterHeight(builder, filterHeight):
|
|
builder.PrependInt32Slot(4, filterHeight, 0)
|
|
|
|
def Pool2DOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(5, fusedActivationFunction, 0)
|
|
|
|
def Pool2DOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class Pool2DOptionsT(object):
|
|
|
|
# Pool2DOptionsT
|
|
def __init__(self):
|
|
self.padding = 0 # type: int
|
|
self.strideW = 0 # type: int
|
|
self.strideH = 0 # type: int
|
|
self.filterWidth = 0 # type: int
|
|
self.filterHeight = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
pool2Doptions = Pool2DOptions()
|
|
pool2Doptions.Init(buf, pos)
|
|
return cls.InitFromObj(pool2Doptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, pool2Doptions):
|
|
x = Pool2DOptionsT()
|
|
x._UnPack(pool2Doptions)
|
|
return x
|
|
|
|
# Pool2DOptionsT
|
|
def _UnPack(self, pool2Doptions):
|
|
if pool2Doptions is None:
|
|
return
|
|
self.padding = pool2Doptions.Padding()
|
|
self.strideW = pool2Doptions.StrideW()
|
|
self.strideH = pool2Doptions.StrideH()
|
|
self.filterWidth = pool2Doptions.FilterWidth()
|
|
self.filterHeight = pool2Doptions.FilterHeight()
|
|
self.fusedActivationFunction = pool2Doptions.FusedActivationFunction()
|
|
|
|
# Pool2DOptionsT
|
|
def Pack(self, builder):
|
|
Pool2DOptionsStart(builder)
|
|
Pool2DOptionsAddPadding(builder, self.padding)
|
|
Pool2DOptionsAddStrideW(builder, self.strideW)
|
|
Pool2DOptionsAddStrideH(builder, self.strideH)
|
|
Pool2DOptionsAddFilterWidth(builder, self.filterWidth)
|
|
Pool2DOptionsAddFilterHeight(builder, self.filterHeight)
|
|
Pool2DOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
pool2Doptions = Pool2DOptionsEnd(builder)
|
|
return pool2Doptions
|
|
|
|
|
|
class DepthwiseConv2DOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DepthwiseConv2DOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDepthwiseConv2DOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DepthwiseConv2DOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DepthwiseConv2DOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# DepthwiseConv2DOptions
|
|
def Padding(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DepthwiseConv2DOptions
|
|
def StrideW(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DepthwiseConv2DOptions
|
|
def StrideH(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DepthwiseConv2DOptions
|
|
def DepthMultiplier(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DepthwiseConv2DOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# DepthwiseConv2DOptions
|
|
def DilationWFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# DepthwiseConv2DOptions
|
|
def DilationHFactor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
def DepthwiseConv2DOptionsStart(builder):
|
|
builder.StartObject(7)
|
|
|
|
def DepthwiseConv2DOptionsAddPadding(builder, padding):
|
|
builder.PrependInt8Slot(0, padding, 0)
|
|
|
|
def DepthwiseConv2DOptionsAddStrideW(builder, strideW):
|
|
builder.PrependInt32Slot(1, strideW, 0)
|
|
|
|
def DepthwiseConv2DOptionsAddStrideH(builder, strideH):
|
|
builder.PrependInt32Slot(2, strideH, 0)
|
|
|
|
def DepthwiseConv2DOptionsAddDepthMultiplier(builder, depthMultiplier):
|
|
builder.PrependInt32Slot(3, depthMultiplier, 0)
|
|
|
|
def DepthwiseConv2DOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(4, fusedActivationFunction, 0)
|
|
|
|
def DepthwiseConv2DOptionsAddDilationWFactor(builder, dilationWFactor):
|
|
builder.PrependInt32Slot(5, dilationWFactor, 1)
|
|
|
|
def DepthwiseConv2DOptionsAddDilationHFactor(builder, dilationHFactor):
|
|
builder.PrependInt32Slot(6, dilationHFactor, 1)
|
|
|
|
def DepthwiseConv2DOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DepthwiseConv2DOptionsT(object):
|
|
|
|
# DepthwiseConv2DOptionsT
|
|
def __init__(self):
|
|
self.padding = 0 # type: int
|
|
self.strideW = 0 # type: int
|
|
self.strideH = 0 # type: int
|
|
self.depthMultiplier = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.dilationWFactor = 1 # type: int
|
|
self.dilationHFactor = 1 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
depthwiseConv2Doptions = DepthwiseConv2DOptions()
|
|
depthwiseConv2Doptions.Init(buf, pos)
|
|
return cls.InitFromObj(depthwiseConv2Doptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, depthwiseConv2Doptions):
|
|
x = DepthwiseConv2DOptionsT()
|
|
x._UnPack(depthwiseConv2Doptions)
|
|
return x
|
|
|
|
# DepthwiseConv2DOptionsT
|
|
def _UnPack(self, depthwiseConv2Doptions):
|
|
if depthwiseConv2Doptions is None:
|
|
return
|
|
self.padding = depthwiseConv2Doptions.Padding()
|
|
self.strideW = depthwiseConv2Doptions.StrideW()
|
|
self.strideH = depthwiseConv2Doptions.StrideH()
|
|
self.depthMultiplier = depthwiseConv2Doptions.DepthMultiplier()
|
|
self.fusedActivationFunction = depthwiseConv2Doptions.FusedActivationFunction()
|
|
self.dilationWFactor = depthwiseConv2Doptions.DilationWFactor()
|
|
self.dilationHFactor = depthwiseConv2Doptions.DilationHFactor()
|
|
|
|
# DepthwiseConv2DOptionsT
|
|
def Pack(self, builder):
|
|
DepthwiseConv2DOptionsStart(builder)
|
|
DepthwiseConv2DOptionsAddPadding(builder, self.padding)
|
|
DepthwiseConv2DOptionsAddStrideW(builder, self.strideW)
|
|
DepthwiseConv2DOptionsAddStrideH(builder, self.strideH)
|
|
DepthwiseConv2DOptionsAddDepthMultiplier(builder, self.depthMultiplier)
|
|
DepthwiseConv2DOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
DepthwiseConv2DOptionsAddDilationWFactor(builder, self.dilationWFactor)
|
|
DepthwiseConv2DOptionsAddDilationHFactor(builder, self.dilationHFactor)
|
|
depthwiseConv2Doptions = DepthwiseConv2DOptionsEnd(builder)
|
|
return depthwiseConv2Doptions
|
|
|
|
|
|
class ConcatEmbeddingsOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ConcatEmbeddingsOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsConcatEmbeddingsOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ConcatEmbeddingsOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def NumChannels(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def NumColumnsPerChannel(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def NumColumnsPerChannelAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def NumColumnsPerChannelLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def NumColumnsPerChannelIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def EmbeddingDimPerChannel(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def EmbeddingDimPerChannelAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def EmbeddingDimPerChannelLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# ConcatEmbeddingsOptions
|
|
def EmbeddingDimPerChannelIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
def ConcatEmbeddingsOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def ConcatEmbeddingsOptionsAddNumChannels(builder, numChannels):
|
|
builder.PrependInt32Slot(0, numChannels, 0)
|
|
|
|
def ConcatEmbeddingsOptionsAddNumColumnsPerChannel(builder, numColumnsPerChannel):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(numColumnsPerChannel), 0)
|
|
|
|
def ConcatEmbeddingsOptionsStartNumColumnsPerChannelVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ConcatEmbeddingsOptionsAddEmbeddingDimPerChannel(builder, embeddingDimPerChannel):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(embeddingDimPerChannel), 0)
|
|
|
|
def ConcatEmbeddingsOptionsStartEmbeddingDimPerChannelVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ConcatEmbeddingsOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class ConcatEmbeddingsOptionsT(object):
|
|
|
|
# ConcatEmbeddingsOptionsT
|
|
def __init__(self):
|
|
self.numChannels = 0 # type: int
|
|
self.numColumnsPerChannel = None # type: List[int]
|
|
self.embeddingDimPerChannel = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
concatEmbeddingsOptions = ConcatEmbeddingsOptions()
|
|
concatEmbeddingsOptions.Init(buf, pos)
|
|
return cls.InitFromObj(concatEmbeddingsOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, concatEmbeddingsOptions):
|
|
x = ConcatEmbeddingsOptionsT()
|
|
x._UnPack(concatEmbeddingsOptions)
|
|
return x
|
|
|
|
# ConcatEmbeddingsOptionsT
|
|
def _UnPack(self, concatEmbeddingsOptions):
|
|
if concatEmbeddingsOptions is None:
|
|
return
|
|
self.numChannels = concatEmbeddingsOptions.NumChannels()
|
|
if not concatEmbeddingsOptions.NumColumnsPerChannelIsNone():
|
|
if np is None:
|
|
self.numColumnsPerChannel = []
|
|
for i in range(concatEmbeddingsOptions.NumColumnsPerChannelLength()):
|
|
self.numColumnsPerChannel.append(concatEmbeddingsOptions.NumColumnsPerChannel(i))
|
|
else:
|
|
self.numColumnsPerChannel = concatEmbeddingsOptions.NumColumnsPerChannelAsNumpy()
|
|
if not concatEmbeddingsOptions.EmbeddingDimPerChannelIsNone():
|
|
if np is None:
|
|
self.embeddingDimPerChannel = []
|
|
for i in range(concatEmbeddingsOptions.EmbeddingDimPerChannelLength()):
|
|
self.embeddingDimPerChannel.append(concatEmbeddingsOptions.EmbeddingDimPerChannel(i))
|
|
else:
|
|
self.embeddingDimPerChannel = concatEmbeddingsOptions.EmbeddingDimPerChannelAsNumpy()
|
|
|
|
# ConcatEmbeddingsOptionsT
|
|
def Pack(self, builder):
|
|
if self.numColumnsPerChannel is not None:
|
|
if np is not None and type(self.numColumnsPerChannel) is np.ndarray:
|
|
numColumnsPerChannel = builder.CreateNumpyVector(self.numColumnsPerChannel)
|
|
else:
|
|
ConcatEmbeddingsOptionsStartNumColumnsPerChannelVector(builder, len(self.numColumnsPerChannel))
|
|
for i in reversed(range(len(self.numColumnsPerChannel))):
|
|
builder.PrependInt32(self.numColumnsPerChannel[i])
|
|
numColumnsPerChannel = builder.EndVector()
|
|
if self.embeddingDimPerChannel is not None:
|
|
if np is not None and type(self.embeddingDimPerChannel) is np.ndarray:
|
|
embeddingDimPerChannel = builder.CreateNumpyVector(self.embeddingDimPerChannel)
|
|
else:
|
|
ConcatEmbeddingsOptionsStartEmbeddingDimPerChannelVector(builder, len(self.embeddingDimPerChannel))
|
|
for i in reversed(range(len(self.embeddingDimPerChannel))):
|
|
builder.PrependInt32(self.embeddingDimPerChannel[i])
|
|
embeddingDimPerChannel = builder.EndVector()
|
|
ConcatEmbeddingsOptionsStart(builder)
|
|
ConcatEmbeddingsOptionsAddNumChannels(builder, self.numChannels)
|
|
if self.numColumnsPerChannel is not None:
|
|
ConcatEmbeddingsOptionsAddNumColumnsPerChannel(builder, numColumnsPerChannel)
|
|
if self.embeddingDimPerChannel is not None:
|
|
ConcatEmbeddingsOptionsAddEmbeddingDimPerChannel(builder, embeddingDimPerChannel)
|
|
concatEmbeddingsOptions = ConcatEmbeddingsOptionsEnd(builder)
|
|
return concatEmbeddingsOptions
|
|
|
|
|
|
class LSHProjectionOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LSHProjectionOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLSHProjectionOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LSHProjectionOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LSHProjectionOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# LSHProjectionOptions
|
|
def Type(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def LSHProjectionOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def LSHProjectionOptionsAddType(builder, type):
|
|
builder.PrependInt8Slot(0, type, 0)
|
|
|
|
def LSHProjectionOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LSHProjectionOptionsT(object):
|
|
|
|
# LSHProjectionOptionsT
|
|
def __init__(self):
|
|
self.type = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
lshprojectionOptions = LSHProjectionOptions()
|
|
lshprojectionOptions.Init(buf, pos)
|
|
return cls.InitFromObj(lshprojectionOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, lshprojectionOptions):
|
|
x = LSHProjectionOptionsT()
|
|
x._UnPack(lshprojectionOptions)
|
|
return x
|
|
|
|
# LSHProjectionOptionsT
|
|
def _UnPack(self, lshprojectionOptions):
|
|
if lshprojectionOptions is None:
|
|
return
|
|
self.type = lshprojectionOptions.Type()
|
|
|
|
# LSHProjectionOptionsT
|
|
def Pack(self, builder):
|
|
LSHProjectionOptionsStart(builder)
|
|
LSHProjectionOptionsAddType(builder, self.type)
|
|
lshprojectionOptions = LSHProjectionOptionsEnd(builder)
|
|
return lshprojectionOptions
|
|
|
|
|
|
class SVDFOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SVDFOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSVDFOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SVDFOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SVDFOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SVDFOptions
|
|
def Rank(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SVDFOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SVDFOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def SVDFOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def SVDFOptionsAddRank(builder, rank):
|
|
builder.PrependInt32Slot(0, rank, 0)
|
|
|
|
def SVDFOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(1, fusedActivationFunction, 0)
|
|
|
|
def SVDFOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(2, asymmetricQuantizeInputs, 0)
|
|
|
|
def SVDFOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SVDFOptionsT(object):
|
|
|
|
# SVDFOptionsT
|
|
def __init__(self):
|
|
self.rank = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
svdfoptions = SVDFOptions()
|
|
svdfoptions.Init(buf, pos)
|
|
return cls.InitFromObj(svdfoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, svdfoptions):
|
|
x = SVDFOptionsT()
|
|
x._UnPack(svdfoptions)
|
|
return x
|
|
|
|
# SVDFOptionsT
|
|
def _UnPack(self, svdfoptions):
|
|
if svdfoptions is None:
|
|
return
|
|
self.rank = svdfoptions.Rank()
|
|
self.fusedActivationFunction = svdfoptions.FusedActivationFunction()
|
|
self.asymmetricQuantizeInputs = svdfoptions.AsymmetricQuantizeInputs()
|
|
|
|
# SVDFOptionsT
|
|
def Pack(self, builder):
|
|
SVDFOptionsStart(builder)
|
|
SVDFOptionsAddRank(builder, self.rank)
|
|
SVDFOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
SVDFOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
svdfoptions = SVDFOptionsEnd(builder)
|
|
return svdfoptions
|
|
|
|
|
|
class RNNOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = RNNOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRNNOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def RNNOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# RNNOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# RNNOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# RNNOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def RNNOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def RNNOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def RNNOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(1, asymmetricQuantizeInputs, 0)
|
|
|
|
def RNNOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class RNNOptionsT(object):
|
|
|
|
# RNNOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
rnnoptions = RNNOptions()
|
|
rnnoptions.Init(buf, pos)
|
|
return cls.InitFromObj(rnnoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, rnnoptions):
|
|
x = RNNOptionsT()
|
|
x._UnPack(rnnoptions)
|
|
return x
|
|
|
|
# RNNOptionsT
|
|
def _UnPack(self, rnnoptions):
|
|
if rnnoptions is None:
|
|
return
|
|
self.fusedActivationFunction = rnnoptions.FusedActivationFunction()
|
|
self.asymmetricQuantizeInputs = rnnoptions.AsymmetricQuantizeInputs()
|
|
|
|
# RNNOptionsT
|
|
def Pack(self, builder):
|
|
RNNOptionsStart(builder)
|
|
RNNOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
RNNOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
rnnoptions = RNNOptionsEnd(builder)
|
|
return rnnoptions
|
|
|
|
|
|
class SequenceRNNOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SequenceRNNOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSequenceRNNOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SequenceRNNOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SequenceRNNOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SequenceRNNOptions
|
|
def TimeMajor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# SequenceRNNOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SequenceRNNOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def SequenceRNNOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def SequenceRNNOptionsAddTimeMajor(builder, timeMajor):
|
|
builder.PrependBoolSlot(0, timeMajor, 0)
|
|
|
|
def SequenceRNNOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(1, fusedActivationFunction, 0)
|
|
|
|
def SequenceRNNOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(2, asymmetricQuantizeInputs, 0)
|
|
|
|
def SequenceRNNOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SequenceRNNOptionsT(object):
|
|
|
|
# SequenceRNNOptionsT
|
|
def __init__(self):
|
|
self.timeMajor = False # type: bool
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
sequenceRnnoptions = SequenceRNNOptions()
|
|
sequenceRnnoptions.Init(buf, pos)
|
|
return cls.InitFromObj(sequenceRnnoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, sequenceRnnoptions):
|
|
x = SequenceRNNOptionsT()
|
|
x._UnPack(sequenceRnnoptions)
|
|
return x
|
|
|
|
# SequenceRNNOptionsT
|
|
def _UnPack(self, sequenceRnnoptions):
|
|
if sequenceRnnoptions is None:
|
|
return
|
|
self.timeMajor = sequenceRnnoptions.TimeMajor()
|
|
self.fusedActivationFunction = sequenceRnnoptions.FusedActivationFunction()
|
|
self.asymmetricQuantizeInputs = sequenceRnnoptions.AsymmetricQuantizeInputs()
|
|
|
|
# SequenceRNNOptionsT
|
|
def Pack(self, builder):
|
|
SequenceRNNOptionsStart(builder)
|
|
SequenceRNNOptionsAddTimeMajor(builder, self.timeMajor)
|
|
SequenceRNNOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
SequenceRNNOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
sequenceRnnoptions = SequenceRNNOptionsEnd(builder)
|
|
return sequenceRnnoptions
|
|
|
|
|
|
class BidirectionalSequenceRNNOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BidirectionalSequenceRNNOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBidirectionalSequenceRNNOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BidirectionalSequenceRNNOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BidirectionalSequenceRNNOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# BidirectionalSequenceRNNOptions
|
|
def TimeMajor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# BidirectionalSequenceRNNOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# BidirectionalSequenceRNNOptions
|
|
def MergeOutputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# BidirectionalSequenceRNNOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def BidirectionalSequenceRNNOptionsStart(builder):
|
|
builder.StartObject(4)
|
|
|
|
def BidirectionalSequenceRNNOptionsAddTimeMajor(builder, timeMajor):
|
|
builder.PrependBoolSlot(0, timeMajor, 0)
|
|
|
|
def BidirectionalSequenceRNNOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(1, fusedActivationFunction, 0)
|
|
|
|
def BidirectionalSequenceRNNOptionsAddMergeOutputs(builder, mergeOutputs):
|
|
builder.PrependBoolSlot(2, mergeOutputs, 0)
|
|
|
|
def BidirectionalSequenceRNNOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(3, asymmetricQuantizeInputs, 0)
|
|
|
|
def BidirectionalSequenceRNNOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BidirectionalSequenceRNNOptionsT(object):
|
|
|
|
# BidirectionalSequenceRNNOptionsT
|
|
def __init__(self):
|
|
self.timeMajor = False # type: bool
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.mergeOutputs = False # type: bool
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
bidirectionalSequenceRnnoptions = BidirectionalSequenceRNNOptions()
|
|
bidirectionalSequenceRnnoptions.Init(buf, pos)
|
|
return cls.InitFromObj(bidirectionalSequenceRnnoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, bidirectionalSequenceRnnoptions):
|
|
x = BidirectionalSequenceRNNOptionsT()
|
|
x._UnPack(bidirectionalSequenceRnnoptions)
|
|
return x
|
|
|
|
# BidirectionalSequenceRNNOptionsT
|
|
def _UnPack(self, bidirectionalSequenceRnnoptions):
|
|
if bidirectionalSequenceRnnoptions is None:
|
|
return
|
|
self.timeMajor = bidirectionalSequenceRnnoptions.TimeMajor()
|
|
self.fusedActivationFunction = bidirectionalSequenceRnnoptions.FusedActivationFunction()
|
|
self.mergeOutputs = bidirectionalSequenceRnnoptions.MergeOutputs()
|
|
self.asymmetricQuantizeInputs = bidirectionalSequenceRnnoptions.AsymmetricQuantizeInputs()
|
|
|
|
# BidirectionalSequenceRNNOptionsT
|
|
def Pack(self, builder):
|
|
BidirectionalSequenceRNNOptionsStart(builder)
|
|
BidirectionalSequenceRNNOptionsAddTimeMajor(builder, self.timeMajor)
|
|
BidirectionalSequenceRNNOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
BidirectionalSequenceRNNOptionsAddMergeOutputs(builder, self.mergeOutputs)
|
|
BidirectionalSequenceRNNOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
bidirectionalSequenceRnnoptions = BidirectionalSequenceRNNOptionsEnd(builder)
|
|
return bidirectionalSequenceRnnoptions
|
|
|
|
|
|
class FullyConnectedOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = FullyConnectedOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsFullyConnectedOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def FullyConnectedOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# FullyConnectedOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# FullyConnectedOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# FullyConnectedOptions
|
|
def WeightsFormat(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# FullyConnectedOptions
|
|
def KeepNumDims(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# FullyConnectedOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# FullyConnectedOptions
|
|
def QuantizedBiasType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def FullyConnectedOptionsStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def FullyConnectedOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def FullyConnectedOptionsAddWeightsFormat(builder, weightsFormat):
|
|
builder.PrependInt8Slot(1, weightsFormat, 0)
|
|
|
|
def FullyConnectedOptionsAddKeepNumDims(builder, keepNumDims):
|
|
builder.PrependBoolSlot(2, keepNumDims, 0)
|
|
|
|
def FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(3, asymmetricQuantizeInputs, 0)
|
|
|
|
def FullyConnectedOptionsAddQuantizedBiasType(builder, quantizedBiasType):
|
|
builder.PrependInt8Slot(4, quantizedBiasType, 0)
|
|
|
|
def FullyConnectedOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class FullyConnectedOptionsT(object):
|
|
|
|
# FullyConnectedOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.weightsFormat = 0 # type: int
|
|
self.keepNumDims = False # type: bool
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
self.quantizedBiasType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
fullyConnectedOptions = FullyConnectedOptions()
|
|
fullyConnectedOptions.Init(buf, pos)
|
|
return cls.InitFromObj(fullyConnectedOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, fullyConnectedOptions):
|
|
x = FullyConnectedOptionsT()
|
|
x._UnPack(fullyConnectedOptions)
|
|
return x
|
|
|
|
# FullyConnectedOptionsT
|
|
def _UnPack(self, fullyConnectedOptions):
|
|
if fullyConnectedOptions is None:
|
|
return
|
|
self.fusedActivationFunction = fullyConnectedOptions.FusedActivationFunction()
|
|
self.weightsFormat = fullyConnectedOptions.WeightsFormat()
|
|
self.keepNumDims = fullyConnectedOptions.KeepNumDims()
|
|
self.asymmetricQuantizeInputs = fullyConnectedOptions.AsymmetricQuantizeInputs()
|
|
self.quantizedBiasType = fullyConnectedOptions.QuantizedBiasType()
|
|
|
|
# FullyConnectedOptionsT
|
|
def Pack(self, builder):
|
|
FullyConnectedOptionsStart(builder)
|
|
FullyConnectedOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
FullyConnectedOptionsAddWeightsFormat(builder, self.weightsFormat)
|
|
FullyConnectedOptionsAddKeepNumDims(builder, self.keepNumDims)
|
|
FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
FullyConnectedOptionsAddQuantizedBiasType(builder, self.quantizedBiasType)
|
|
fullyConnectedOptions = FullyConnectedOptionsEnd(builder)
|
|
return fullyConnectedOptions
|
|
|
|
|
|
class SoftmaxOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SoftmaxOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSoftmaxOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SoftmaxOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SoftmaxOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SoftmaxOptions
|
|
def Beta(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
def SoftmaxOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SoftmaxOptionsAddBeta(builder, beta):
|
|
builder.PrependFloat32Slot(0, beta, 0.0)
|
|
|
|
def SoftmaxOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SoftmaxOptionsT(object):
|
|
|
|
# SoftmaxOptionsT
|
|
def __init__(self):
|
|
self.beta = 0.0 # type: float
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
softmaxOptions = SoftmaxOptions()
|
|
softmaxOptions.Init(buf, pos)
|
|
return cls.InitFromObj(softmaxOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, softmaxOptions):
|
|
x = SoftmaxOptionsT()
|
|
x._UnPack(softmaxOptions)
|
|
return x
|
|
|
|
# SoftmaxOptionsT
|
|
def _UnPack(self, softmaxOptions):
|
|
if softmaxOptions is None:
|
|
return
|
|
self.beta = softmaxOptions.Beta()
|
|
|
|
# SoftmaxOptionsT
|
|
def Pack(self, builder):
|
|
SoftmaxOptionsStart(builder)
|
|
SoftmaxOptionsAddBeta(builder, self.beta)
|
|
softmaxOptions = SoftmaxOptionsEnd(builder)
|
|
return softmaxOptions
|
|
|
|
|
|
class ConcatenationOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ConcatenationOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsConcatenationOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ConcatenationOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ConcatenationOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ConcatenationOptions
|
|
def Axis(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# ConcatenationOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ConcatenationOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def ConcatenationOptionsAddAxis(builder, axis):
|
|
builder.PrependInt32Slot(0, axis, 0)
|
|
|
|
def ConcatenationOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(1, fusedActivationFunction, 0)
|
|
|
|
def ConcatenationOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ConcatenationOptionsT(object):
|
|
|
|
# ConcatenationOptionsT
|
|
def __init__(self):
|
|
self.axis = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
concatenationOptions = ConcatenationOptions()
|
|
concatenationOptions.Init(buf, pos)
|
|
return cls.InitFromObj(concatenationOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, concatenationOptions):
|
|
x = ConcatenationOptionsT()
|
|
x._UnPack(concatenationOptions)
|
|
return x
|
|
|
|
# ConcatenationOptionsT
|
|
def _UnPack(self, concatenationOptions):
|
|
if concatenationOptions is None:
|
|
return
|
|
self.axis = concatenationOptions.Axis()
|
|
self.fusedActivationFunction = concatenationOptions.FusedActivationFunction()
|
|
|
|
# ConcatenationOptionsT
|
|
def Pack(self, builder):
|
|
ConcatenationOptionsStart(builder)
|
|
ConcatenationOptionsAddAxis(builder, self.axis)
|
|
ConcatenationOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
concatenationOptions = ConcatenationOptionsEnd(builder)
|
|
return concatenationOptions
|
|
|
|
|
|
class AddOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = AddOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsAddOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def AddOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# AddOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# AddOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# AddOptions
|
|
def PotScaleInt16(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return True
|
|
|
|
def AddOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def AddOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def AddOptionsAddPotScaleInt16(builder, potScaleInt16):
|
|
builder.PrependBoolSlot(1, potScaleInt16, 1)
|
|
|
|
def AddOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class AddOptionsT(object):
|
|
|
|
# AddOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.potScaleInt16 = True # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
addOptions = AddOptions()
|
|
addOptions.Init(buf, pos)
|
|
return cls.InitFromObj(addOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, addOptions):
|
|
x = AddOptionsT()
|
|
x._UnPack(addOptions)
|
|
return x
|
|
|
|
# AddOptionsT
|
|
def _UnPack(self, addOptions):
|
|
if addOptions is None:
|
|
return
|
|
self.fusedActivationFunction = addOptions.FusedActivationFunction()
|
|
self.potScaleInt16 = addOptions.PotScaleInt16()
|
|
|
|
# AddOptionsT
|
|
def Pack(self, builder):
|
|
AddOptionsStart(builder)
|
|
AddOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
AddOptionsAddPotScaleInt16(builder, self.potScaleInt16)
|
|
addOptions = AddOptionsEnd(builder)
|
|
return addOptions
|
|
|
|
|
|
class MulOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = MulOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMulOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MulOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# MulOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# MulOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def MulOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def MulOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def MulOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MulOptionsT(object):
|
|
|
|
# MulOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
mulOptions = MulOptions()
|
|
mulOptions.Init(buf, pos)
|
|
return cls.InitFromObj(mulOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, mulOptions):
|
|
x = MulOptionsT()
|
|
x._UnPack(mulOptions)
|
|
return x
|
|
|
|
# MulOptionsT
|
|
def _UnPack(self, mulOptions):
|
|
if mulOptions is None:
|
|
return
|
|
self.fusedActivationFunction = mulOptions.FusedActivationFunction()
|
|
|
|
# MulOptionsT
|
|
def Pack(self, builder):
|
|
MulOptionsStart(builder)
|
|
MulOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
mulOptions = MulOptionsEnd(builder)
|
|
return mulOptions
|
|
|
|
|
|
class L2NormOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = L2NormOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsL2NormOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def L2NormOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# L2NormOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# L2NormOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def L2NormOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def L2NormOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def L2NormOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class L2NormOptionsT(object):
|
|
|
|
# L2NormOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
l2NormOptions = L2NormOptions()
|
|
l2NormOptions.Init(buf, pos)
|
|
return cls.InitFromObj(l2NormOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, l2NormOptions):
|
|
x = L2NormOptionsT()
|
|
x._UnPack(l2NormOptions)
|
|
return x
|
|
|
|
# L2NormOptionsT
|
|
def _UnPack(self, l2NormOptions):
|
|
if l2NormOptions is None:
|
|
return
|
|
self.fusedActivationFunction = l2NormOptions.FusedActivationFunction()
|
|
|
|
# L2NormOptionsT
|
|
def Pack(self, builder):
|
|
L2NormOptionsStart(builder)
|
|
L2NormOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
l2NormOptions = L2NormOptionsEnd(builder)
|
|
return l2NormOptions
|
|
|
|
|
|
class LocalResponseNormalizationOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LocalResponseNormalizationOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLocalResponseNormalizationOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LocalResponseNormalizationOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LocalResponseNormalizationOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# LocalResponseNormalizationOptions
|
|
def Radius(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# LocalResponseNormalizationOptions
|
|
def Bias(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# LocalResponseNormalizationOptions
|
|
def Alpha(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# LocalResponseNormalizationOptions
|
|
def Beta(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
def LocalResponseNormalizationOptionsStart(builder):
|
|
builder.StartObject(4)
|
|
|
|
def LocalResponseNormalizationOptionsAddRadius(builder, radius):
|
|
builder.PrependInt32Slot(0, radius, 0)
|
|
|
|
def LocalResponseNormalizationOptionsAddBias(builder, bias):
|
|
builder.PrependFloat32Slot(1, bias, 0.0)
|
|
|
|
def LocalResponseNormalizationOptionsAddAlpha(builder, alpha):
|
|
builder.PrependFloat32Slot(2, alpha, 0.0)
|
|
|
|
def LocalResponseNormalizationOptionsAddBeta(builder, beta):
|
|
builder.PrependFloat32Slot(3, beta, 0.0)
|
|
|
|
def LocalResponseNormalizationOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LocalResponseNormalizationOptionsT(object):
|
|
|
|
# LocalResponseNormalizationOptionsT
|
|
def __init__(self):
|
|
self.radius = 0 # type: int
|
|
self.bias = 0.0 # type: float
|
|
self.alpha = 0.0 # type: float
|
|
self.beta = 0.0 # type: float
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
localResponseNormalizationOptions = LocalResponseNormalizationOptions()
|
|
localResponseNormalizationOptions.Init(buf, pos)
|
|
return cls.InitFromObj(localResponseNormalizationOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, localResponseNormalizationOptions):
|
|
x = LocalResponseNormalizationOptionsT()
|
|
x._UnPack(localResponseNormalizationOptions)
|
|
return x
|
|
|
|
# LocalResponseNormalizationOptionsT
|
|
def _UnPack(self, localResponseNormalizationOptions):
|
|
if localResponseNormalizationOptions is None:
|
|
return
|
|
self.radius = localResponseNormalizationOptions.Radius()
|
|
self.bias = localResponseNormalizationOptions.Bias()
|
|
self.alpha = localResponseNormalizationOptions.Alpha()
|
|
self.beta = localResponseNormalizationOptions.Beta()
|
|
|
|
# LocalResponseNormalizationOptionsT
|
|
def Pack(self, builder):
|
|
LocalResponseNormalizationOptionsStart(builder)
|
|
LocalResponseNormalizationOptionsAddRadius(builder, self.radius)
|
|
LocalResponseNormalizationOptionsAddBias(builder, self.bias)
|
|
LocalResponseNormalizationOptionsAddAlpha(builder, self.alpha)
|
|
LocalResponseNormalizationOptionsAddBeta(builder, self.beta)
|
|
localResponseNormalizationOptions = LocalResponseNormalizationOptionsEnd(builder)
|
|
return localResponseNormalizationOptions
|
|
|
|
|
|
class LSTMOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LSTMOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLSTMOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LSTMOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LSTMOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# LSTMOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# LSTMOptions
|
|
def CellClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# LSTMOptions
|
|
def ProjClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# LSTMOptions
|
|
def KernelType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# LSTMOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def LSTMOptionsStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def LSTMOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def LSTMOptionsAddCellClip(builder, cellClip):
|
|
builder.PrependFloat32Slot(1, cellClip, 0.0)
|
|
|
|
def LSTMOptionsAddProjClip(builder, projClip):
|
|
builder.PrependFloat32Slot(2, projClip, 0.0)
|
|
|
|
def LSTMOptionsAddKernelType(builder, kernelType):
|
|
builder.PrependInt8Slot(3, kernelType, 0)
|
|
|
|
def LSTMOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(4, asymmetricQuantizeInputs, 0)
|
|
|
|
def LSTMOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LSTMOptionsT(object):
|
|
|
|
# LSTMOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.cellClip = 0.0 # type: float
|
|
self.projClip = 0.0 # type: float
|
|
self.kernelType = 0 # type: int
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
lstmoptions = LSTMOptions()
|
|
lstmoptions.Init(buf, pos)
|
|
return cls.InitFromObj(lstmoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, lstmoptions):
|
|
x = LSTMOptionsT()
|
|
x._UnPack(lstmoptions)
|
|
return x
|
|
|
|
# LSTMOptionsT
|
|
def _UnPack(self, lstmoptions):
|
|
if lstmoptions is None:
|
|
return
|
|
self.fusedActivationFunction = lstmoptions.FusedActivationFunction()
|
|
self.cellClip = lstmoptions.CellClip()
|
|
self.projClip = lstmoptions.ProjClip()
|
|
self.kernelType = lstmoptions.KernelType()
|
|
self.asymmetricQuantizeInputs = lstmoptions.AsymmetricQuantizeInputs()
|
|
|
|
# LSTMOptionsT
|
|
def Pack(self, builder):
|
|
LSTMOptionsStart(builder)
|
|
LSTMOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
LSTMOptionsAddCellClip(builder, self.cellClip)
|
|
LSTMOptionsAddProjClip(builder, self.projClip)
|
|
LSTMOptionsAddKernelType(builder, self.kernelType)
|
|
LSTMOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
lstmoptions = LSTMOptionsEnd(builder)
|
|
return lstmoptions
|
|
|
|
|
|
class UnidirectionalSequenceLSTMOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnidirectionalSequenceLSTMOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnidirectionalSequenceLSTMOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnidirectionalSequenceLSTMOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def CellClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def ProjClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def TimeMajor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# UnidirectionalSequenceLSTMOptions
|
|
def DiagonalRecurrentTensors(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def UnidirectionalSequenceLSTMOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddCellClip(builder, cellClip):
|
|
builder.PrependFloat32Slot(1, cellClip, 0.0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddProjClip(builder, projClip):
|
|
builder.PrependFloat32Slot(2, projClip, 0.0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddTimeMajor(builder, timeMajor):
|
|
builder.PrependBoolSlot(3, timeMajor, 0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(4, asymmetricQuantizeInputs, 0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsAddDiagonalRecurrentTensors(builder, diagonalRecurrentTensors):
|
|
builder.PrependBoolSlot(5, diagonalRecurrentTensors, 0)
|
|
|
|
def UnidirectionalSequenceLSTMOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnidirectionalSequenceLSTMOptionsT(object):
|
|
|
|
# UnidirectionalSequenceLSTMOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.cellClip = 0.0 # type: float
|
|
self.projClip = 0.0 # type: float
|
|
self.timeMajor = False # type: bool
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
self.diagonalRecurrentTensors = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unidirectionalSequenceLstmoptions = UnidirectionalSequenceLSTMOptions()
|
|
unidirectionalSequenceLstmoptions.Init(buf, pos)
|
|
return cls.InitFromObj(unidirectionalSequenceLstmoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unidirectionalSequenceLstmoptions):
|
|
x = UnidirectionalSequenceLSTMOptionsT()
|
|
x._UnPack(unidirectionalSequenceLstmoptions)
|
|
return x
|
|
|
|
# UnidirectionalSequenceLSTMOptionsT
|
|
def _UnPack(self, unidirectionalSequenceLstmoptions):
|
|
if unidirectionalSequenceLstmoptions is None:
|
|
return
|
|
self.fusedActivationFunction = unidirectionalSequenceLstmoptions.FusedActivationFunction()
|
|
self.cellClip = unidirectionalSequenceLstmoptions.CellClip()
|
|
self.projClip = unidirectionalSequenceLstmoptions.ProjClip()
|
|
self.timeMajor = unidirectionalSequenceLstmoptions.TimeMajor()
|
|
self.asymmetricQuantizeInputs = unidirectionalSequenceLstmoptions.AsymmetricQuantizeInputs()
|
|
self.diagonalRecurrentTensors = unidirectionalSequenceLstmoptions.DiagonalRecurrentTensors()
|
|
|
|
# UnidirectionalSequenceLSTMOptionsT
|
|
def Pack(self, builder):
|
|
UnidirectionalSequenceLSTMOptionsStart(builder)
|
|
UnidirectionalSequenceLSTMOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
UnidirectionalSequenceLSTMOptionsAddCellClip(builder, self.cellClip)
|
|
UnidirectionalSequenceLSTMOptionsAddProjClip(builder, self.projClip)
|
|
UnidirectionalSequenceLSTMOptionsAddTimeMajor(builder, self.timeMajor)
|
|
UnidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
UnidirectionalSequenceLSTMOptionsAddDiagonalRecurrentTensors(builder, self.diagonalRecurrentTensors)
|
|
unidirectionalSequenceLstmoptions = UnidirectionalSequenceLSTMOptionsEnd(builder)
|
|
return unidirectionalSequenceLstmoptions
|
|
|
|
|
|
class BidirectionalSequenceLSTMOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BidirectionalSequenceLSTMOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBidirectionalSequenceLSTMOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BidirectionalSequenceLSTMOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def CellClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def ProjClip(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def MergeOutputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def TimeMajor(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return True
|
|
|
|
# BidirectionalSequenceLSTMOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def BidirectionalSequenceLSTMOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddCellClip(builder, cellClip):
|
|
builder.PrependFloat32Slot(1, cellClip, 0.0)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddProjClip(builder, projClip):
|
|
builder.PrependFloat32Slot(2, projClip, 0.0)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddMergeOutputs(builder, mergeOutputs):
|
|
builder.PrependBoolSlot(3, mergeOutputs, 0)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddTimeMajor(builder, timeMajor):
|
|
builder.PrependBoolSlot(4, timeMajor, 1)
|
|
|
|
def BidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(5, asymmetricQuantizeInputs, 0)
|
|
|
|
def BidirectionalSequenceLSTMOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BidirectionalSequenceLSTMOptionsT(object):
|
|
|
|
# BidirectionalSequenceLSTMOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.cellClip = 0.0 # type: float
|
|
self.projClip = 0.0 # type: float
|
|
self.mergeOutputs = False # type: bool
|
|
self.timeMajor = True # type: bool
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
bidirectionalSequenceLstmoptions = BidirectionalSequenceLSTMOptions()
|
|
bidirectionalSequenceLstmoptions.Init(buf, pos)
|
|
return cls.InitFromObj(bidirectionalSequenceLstmoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, bidirectionalSequenceLstmoptions):
|
|
x = BidirectionalSequenceLSTMOptionsT()
|
|
x._UnPack(bidirectionalSequenceLstmoptions)
|
|
return x
|
|
|
|
# BidirectionalSequenceLSTMOptionsT
|
|
def _UnPack(self, bidirectionalSequenceLstmoptions):
|
|
if bidirectionalSequenceLstmoptions is None:
|
|
return
|
|
self.fusedActivationFunction = bidirectionalSequenceLstmoptions.FusedActivationFunction()
|
|
self.cellClip = bidirectionalSequenceLstmoptions.CellClip()
|
|
self.projClip = bidirectionalSequenceLstmoptions.ProjClip()
|
|
self.mergeOutputs = bidirectionalSequenceLstmoptions.MergeOutputs()
|
|
self.timeMajor = bidirectionalSequenceLstmoptions.TimeMajor()
|
|
self.asymmetricQuantizeInputs = bidirectionalSequenceLstmoptions.AsymmetricQuantizeInputs()
|
|
|
|
# BidirectionalSequenceLSTMOptionsT
|
|
def Pack(self, builder):
|
|
BidirectionalSequenceLSTMOptionsStart(builder)
|
|
BidirectionalSequenceLSTMOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
BidirectionalSequenceLSTMOptionsAddCellClip(builder, self.cellClip)
|
|
BidirectionalSequenceLSTMOptionsAddProjClip(builder, self.projClip)
|
|
BidirectionalSequenceLSTMOptionsAddMergeOutputs(builder, self.mergeOutputs)
|
|
BidirectionalSequenceLSTMOptionsAddTimeMajor(builder, self.timeMajor)
|
|
BidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
bidirectionalSequenceLstmoptions = BidirectionalSequenceLSTMOptionsEnd(builder)
|
|
return bidirectionalSequenceLstmoptions
|
|
|
|
|
|
class ResizeBilinearOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ResizeBilinearOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsResizeBilinearOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ResizeBilinearOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ResizeBilinearOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ResizeBilinearOptions
|
|
def AlignCorners(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# ResizeBilinearOptions
|
|
def HalfPixelCenters(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def ResizeBilinearOptionsStart(builder):
|
|
builder.StartObject(4)
|
|
|
|
def ResizeBilinearOptionsAddAlignCorners(builder, alignCorners):
|
|
builder.PrependBoolSlot(2, alignCorners, 0)
|
|
|
|
def ResizeBilinearOptionsAddHalfPixelCenters(builder, halfPixelCenters):
|
|
builder.PrependBoolSlot(3, halfPixelCenters, 0)
|
|
|
|
def ResizeBilinearOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ResizeBilinearOptionsT(object):
|
|
|
|
# ResizeBilinearOptionsT
|
|
def __init__(self):
|
|
self.alignCorners = False # type: bool
|
|
self.halfPixelCenters = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
resizeBilinearOptions = ResizeBilinearOptions()
|
|
resizeBilinearOptions.Init(buf, pos)
|
|
return cls.InitFromObj(resizeBilinearOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, resizeBilinearOptions):
|
|
x = ResizeBilinearOptionsT()
|
|
x._UnPack(resizeBilinearOptions)
|
|
return x
|
|
|
|
# ResizeBilinearOptionsT
|
|
def _UnPack(self, resizeBilinearOptions):
|
|
if resizeBilinearOptions is None:
|
|
return
|
|
self.alignCorners = resizeBilinearOptions.AlignCorners()
|
|
self.halfPixelCenters = resizeBilinearOptions.HalfPixelCenters()
|
|
|
|
# ResizeBilinearOptionsT
|
|
def Pack(self, builder):
|
|
ResizeBilinearOptionsStart(builder)
|
|
ResizeBilinearOptionsAddAlignCorners(builder, self.alignCorners)
|
|
ResizeBilinearOptionsAddHalfPixelCenters(builder, self.halfPixelCenters)
|
|
resizeBilinearOptions = ResizeBilinearOptionsEnd(builder)
|
|
return resizeBilinearOptions
|
|
|
|
|
|
class ResizeNearestNeighborOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ResizeNearestNeighborOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsResizeNearestNeighborOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ResizeNearestNeighborOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ResizeNearestNeighborOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ResizeNearestNeighborOptions
|
|
def AlignCorners(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# ResizeNearestNeighborOptions
|
|
def HalfPixelCenters(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def ResizeNearestNeighborOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def ResizeNearestNeighborOptionsAddAlignCorners(builder, alignCorners):
|
|
builder.PrependBoolSlot(0, alignCorners, 0)
|
|
|
|
def ResizeNearestNeighborOptionsAddHalfPixelCenters(builder, halfPixelCenters):
|
|
builder.PrependBoolSlot(1, halfPixelCenters, 0)
|
|
|
|
def ResizeNearestNeighborOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ResizeNearestNeighborOptionsT(object):
|
|
|
|
# ResizeNearestNeighborOptionsT
|
|
def __init__(self):
|
|
self.alignCorners = False # type: bool
|
|
self.halfPixelCenters = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
resizeNearestNeighborOptions = ResizeNearestNeighborOptions()
|
|
resizeNearestNeighborOptions.Init(buf, pos)
|
|
return cls.InitFromObj(resizeNearestNeighborOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, resizeNearestNeighborOptions):
|
|
x = ResizeNearestNeighborOptionsT()
|
|
x._UnPack(resizeNearestNeighborOptions)
|
|
return x
|
|
|
|
# ResizeNearestNeighborOptionsT
|
|
def _UnPack(self, resizeNearestNeighborOptions):
|
|
if resizeNearestNeighborOptions is None:
|
|
return
|
|
self.alignCorners = resizeNearestNeighborOptions.AlignCorners()
|
|
self.halfPixelCenters = resizeNearestNeighborOptions.HalfPixelCenters()
|
|
|
|
# ResizeNearestNeighborOptionsT
|
|
def Pack(self, builder):
|
|
ResizeNearestNeighborOptionsStart(builder)
|
|
ResizeNearestNeighborOptionsAddAlignCorners(builder, self.alignCorners)
|
|
ResizeNearestNeighborOptionsAddHalfPixelCenters(builder, self.halfPixelCenters)
|
|
resizeNearestNeighborOptions = ResizeNearestNeighborOptionsEnd(builder)
|
|
return resizeNearestNeighborOptions
|
|
|
|
|
|
class CallOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CallOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCallOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CallOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CallOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# CallOptions
|
|
def Subgraph(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def CallOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def CallOptionsAddSubgraph(builder, subgraph):
|
|
builder.PrependUint32Slot(0, subgraph, 0)
|
|
|
|
def CallOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class CallOptionsT(object):
|
|
|
|
# CallOptionsT
|
|
def __init__(self):
|
|
self.subgraph = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
callOptions = CallOptions()
|
|
callOptions.Init(buf, pos)
|
|
return cls.InitFromObj(callOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, callOptions):
|
|
x = CallOptionsT()
|
|
x._UnPack(callOptions)
|
|
return x
|
|
|
|
# CallOptionsT
|
|
def _UnPack(self, callOptions):
|
|
if callOptions is None:
|
|
return
|
|
self.subgraph = callOptions.Subgraph()
|
|
|
|
# CallOptionsT
|
|
def Pack(self, builder):
|
|
CallOptionsStart(builder)
|
|
CallOptionsAddSubgraph(builder, self.subgraph)
|
|
callOptions = CallOptionsEnd(builder)
|
|
return callOptions
|
|
|
|
|
|
class PadOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = PadOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsPadOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def PadOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# PadOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def PadOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def PadOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class PadOptionsT(object):
|
|
|
|
# PadOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
padOptions = PadOptions()
|
|
padOptions.Init(buf, pos)
|
|
return cls.InitFromObj(padOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, padOptions):
|
|
x = PadOptionsT()
|
|
x._UnPack(padOptions)
|
|
return x
|
|
|
|
# PadOptionsT
|
|
def _UnPack(self, padOptions):
|
|
if padOptions is None:
|
|
return
|
|
|
|
# PadOptionsT
|
|
def Pack(self, builder):
|
|
PadOptionsStart(builder)
|
|
padOptions = PadOptionsEnd(builder)
|
|
return padOptions
|
|
|
|
|
|
class PadV2Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = PadV2Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsPadV2Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def PadV2OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# PadV2Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def PadV2OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def PadV2OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class PadV2OptionsT(object):
|
|
|
|
# PadV2OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
padV2Options = PadV2Options()
|
|
padV2Options.Init(buf, pos)
|
|
return cls.InitFromObj(padV2Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, padV2Options):
|
|
x = PadV2OptionsT()
|
|
x._UnPack(padV2Options)
|
|
return x
|
|
|
|
# PadV2OptionsT
|
|
def _UnPack(self, padV2Options):
|
|
if padV2Options is None:
|
|
return
|
|
|
|
# PadV2OptionsT
|
|
def Pack(self, builder):
|
|
PadV2OptionsStart(builder)
|
|
padV2Options = PadV2OptionsEnd(builder)
|
|
return padV2Options
|
|
|
|
|
|
class ReshapeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReshapeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReshapeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReshapeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReshapeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ReshapeOptions
|
|
def NewShape(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# ReshapeOptions
|
|
def NewShapeAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# ReshapeOptions
|
|
def NewShapeLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# ReshapeOptions
|
|
def NewShapeIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def ReshapeOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ReshapeOptionsAddNewShape(builder, newShape):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(newShape), 0)
|
|
|
|
def ReshapeOptionsStartNewShapeVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ReshapeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class ReshapeOptionsT(object):
|
|
|
|
# ReshapeOptionsT
|
|
def __init__(self):
|
|
self.newShape = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
reshapeOptions = ReshapeOptions()
|
|
reshapeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(reshapeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, reshapeOptions):
|
|
x = ReshapeOptionsT()
|
|
x._UnPack(reshapeOptions)
|
|
return x
|
|
|
|
# ReshapeOptionsT
|
|
def _UnPack(self, reshapeOptions):
|
|
if reshapeOptions is None:
|
|
return
|
|
if not reshapeOptions.NewShapeIsNone():
|
|
if np is None:
|
|
self.newShape = []
|
|
for i in range(reshapeOptions.NewShapeLength()):
|
|
self.newShape.append(reshapeOptions.NewShape(i))
|
|
else:
|
|
self.newShape = reshapeOptions.NewShapeAsNumpy()
|
|
|
|
# ReshapeOptionsT
|
|
def Pack(self, builder):
|
|
if self.newShape is not None:
|
|
if np is not None and type(self.newShape) is np.ndarray:
|
|
newShape = builder.CreateNumpyVector(self.newShape)
|
|
else:
|
|
ReshapeOptionsStartNewShapeVector(builder, len(self.newShape))
|
|
for i in reversed(range(len(self.newShape))):
|
|
builder.PrependInt32(self.newShape[i])
|
|
newShape = builder.EndVector()
|
|
ReshapeOptionsStart(builder)
|
|
if self.newShape is not None:
|
|
ReshapeOptionsAddNewShape(builder, newShape)
|
|
reshapeOptions = ReshapeOptionsEnd(builder)
|
|
return reshapeOptions
|
|
|
|
|
|
class SpaceToBatchNDOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SpaceToBatchNDOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSpaceToBatchNDOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SpaceToBatchNDOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SpaceToBatchNDOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SpaceToBatchNDOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SpaceToBatchNDOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SpaceToBatchNDOptionsT(object):
|
|
|
|
# SpaceToBatchNDOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
spaceToBatchNdoptions = SpaceToBatchNDOptions()
|
|
spaceToBatchNdoptions.Init(buf, pos)
|
|
return cls.InitFromObj(spaceToBatchNdoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, spaceToBatchNdoptions):
|
|
x = SpaceToBatchNDOptionsT()
|
|
x._UnPack(spaceToBatchNdoptions)
|
|
return x
|
|
|
|
# SpaceToBatchNDOptionsT
|
|
def _UnPack(self, spaceToBatchNdoptions):
|
|
if spaceToBatchNdoptions is None:
|
|
return
|
|
|
|
# SpaceToBatchNDOptionsT
|
|
def Pack(self, builder):
|
|
SpaceToBatchNDOptionsStart(builder)
|
|
spaceToBatchNdoptions = SpaceToBatchNDOptionsEnd(builder)
|
|
return spaceToBatchNdoptions
|
|
|
|
|
|
class BatchToSpaceNDOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BatchToSpaceNDOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBatchToSpaceNDOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BatchToSpaceNDOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BatchToSpaceNDOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def BatchToSpaceNDOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def BatchToSpaceNDOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BatchToSpaceNDOptionsT(object):
|
|
|
|
# BatchToSpaceNDOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
batchToSpaceNdoptions = BatchToSpaceNDOptions()
|
|
batchToSpaceNdoptions.Init(buf, pos)
|
|
return cls.InitFromObj(batchToSpaceNdoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, batchToSpaceNdoptions):
|
|
x = BatchToSpaceNDOptionsT()
|
|
x._UnPack(batchToSpaceNdoptions)
|
|
return x
|
|
|
|
# BatchToSpaceNDOptionsT
|
|
def _UnPack(self, batchToSpaceNdoptions):
|
|
if batchToSpaceNdoptions is None:
|
|
return
|
|
|
|
# BatchToSpaceNDOptionsT
|
|
def Pack(self, builder):
|
|
BatchToSpaceNDOptionsStart(builder)
|
|
batchToSpaceNdoptions = BatchToSpaceNDOptionsEnd(builder)
|
|
return batchToSpaceNdoptions
|
|
|
|
|
|
class SkipGramOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SkipGramOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSkipGramOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SkipGramOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SkipGramOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SkipGramOptions
|
|
def NgramSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SkipGramOptions
|
|
def MaxSkipSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SkipGramOptions
|
|
def IncludeAllNgrams(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def SkipGramOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def SkipGramOptionsAddNgramSize(builder, ngramSize):
|
|
builder.PrependInt32Slot(0, ngramSize, 0)
|
|
|
|
def SkipGramOptionsAddMaxSkipSize(builder, maxSkipSize):
|
|
builder.PrependInt32Slot(1, maxSkipSize, 0)
|
|
|
|
def SkipGramOptionsAddIncludeAllNgrams(builder, includeAllNgrams):
|
|
builder.PrependBoolSlot(2, includeAllNgrams, 0)
|
|
|
|
def SkipGramOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SkipGramOptionsT(object):
|
|
|
|
# SkipGramOptionsT
|
|
def __init__(self):
|
|
self.ngramSize = 0 # type: int
|
|
self.maxSkipSize = 0 # type: int
|
|
self.includeAllNgrams = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
skipGramOptions = SkipGramOptions()
|
|
skipGramOptions.Init(buf, pos)
|
|
return cls.InitFromObj(skipGramOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, skipGramOptions):
|
|
x = SkipGramOptionsT()
|
|
x._UnPack(skipGramOptions)
|
|
return x
|
|
|
|
# SkipGramOptionsT
|
|
def _UnPack(self, skipGramOptions):
|
|
if skipGramOptions is None:
|
|
return
|
|
self.ngramSize = skipGramOptions.NgramSize()
|
|
self.maxSkipSize = skipGramOptions.MaxSkipSize()
|
|
self.includeAllNgrams = skipGramOptions.IncludeAllNgrams()
|
|
|
|
# SkipGramOptionsT
|
|
def Pack(self, builder):
|
|
SkipGramOptionsStart(builder)
|
|
SkipGramOptionsAddNgramSize(builder, self.ngramSize)
|
|
SkipGramOptionsAddMaxSkipSize(builder, self.maxSkipSize)
|
|
SkipGramOptionsAddIncludeAllNgrams(builder, self.includeAllNgrams)
|
|
skipGramOptions = SkipGramOptionsEnd(builder)
|
|
return skipGramOptions
|
|
|
|
|
|
class SpaceToDepthOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SpaceToDepthOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSpaceToDepthOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SpaceToDepthOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SpaceToDepthOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SpaceToDepthOptions
|
|
def BlockSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def SpaceToDepthOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SpaceToDepthOptionsAddBlockSize(builder, blockSize):
|
|
builder.PrependInt32Slot(0, blockSize, 0)
|
|
|
|
def SpaceToDepthOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SpaceToDepthOptionsT(object):
|
|
|
|
# SpaceToDepthOptionsT
|
|
def __init__(self):
|
|
self.blockSize = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
spaceToDepthOptions = SpaceToDepthOptions()
|
|
spaceToDepthOptions.Init(buf, pos)
|
|
return cls.InitFromObj(spaceToDepthOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, spaceToDepthOptions):
|
|
x = SpaceToDepthOptionsT()
|
|
x._UnPack(spaceToDepthOptions)
|
|
return x
|
|
|
|
# SpaceToDepthOptionsT
|
|
def _UnPack(self, spaceToDepthOptions):
|
|
if spaceToDepthOptions is None:
|
|
return
|
|
self.blockSize = spaceToDepthOptions.BlockSize()
|
|
|
|
# SpaceToDepthOptionsT
|
|
def Pack(self, builder):
|
|
SpaceToDepthOptionsStart(builder)
|
|
SpaceToDepthOptionsAddBlockSize(builder, self.blockSize)
|
|
spaceToDepthOptions = SpaceToDepthOptionsEnd(builder)
|
|
return spaceToDepthOptions
|
|
|
|
|
|
class DepthToSpaceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DepthToSpaceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDepthToSpaceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DepthToSpaceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DepthToSpaceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# DepthToSpaceOptions
|
|
def BlockSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def DepthToSpaceOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def DepthToSpaceOptionsAddBlockSize(builder, blockSize):
|
|
builder.PrependInt32Slot(0, blockSize, 0)
|
|
|
|
def DepthToSpaceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DepthToSpaceOptionsT(object):
|
|
|
|
# DepthToSpaceOptionsT
|
|
def __init__(self):
|
|
self.blockSize = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
depthToSpaceOptions = DepthToSpaceOptions()
|
|
depthToSpaceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(depthToSpaceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, depthToSpaceOptions):
|
|
x = DepthToSpaceOptionsT()
|
|
x._UnPack(depthToSpaceOptions)
|
|
return x
|
|
|
|
# DepthToSpaceOptionsT
|
|
def _UnPack(self, depthToSpaceOptions):
|
|
if depthToSpaceOptions is None:
|
|
return
|
|
self.blockSize = depthToSpaceOptions.BlockSize()
|
|
|
|
# DepthToSpaceOptionsT
|
|
def Pack(self, builder):
|
|
DepthToSpaceOptionsStart(builder)
|
|
DepthToSpaceOptionsAddBlockSize(builder, self.blockSize)
|
|
depthToSpaceOptions = DepthToSpaceOptionsEnd(builder)
|
|
return depthToSpaceOptions
|
|
|
|
|
|
class SubOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SubOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSubOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SubOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SubOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SubOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# SubOptions
|
|
def PotScaleInt16(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return True
|
|
|
|
def SubOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def SubOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def SubOptionsAddPotScaleInt16(builder, potScaleInt16):
|
|
builder.PrependBoolSlot(1, potScaleInt16, 1)
|
|
|
|
def SubOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SubOptionsT(object):
|
|
|
|
# SubOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.potScaleInt16 = True # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
subOptions = SubOptions()
|
|
subOptions.Init(buf, pos)
|
|
return cls.InitFromObj(subOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, subOptions):
|
|
x = SubOptionsT()
|
|
x._UnPack(subOptions)
|
|
return x
|
|
|
|
# SubOptionsT
|
|
def _UnPack(self, subOptions):
|
|
if subOptions is None:
|
|
return
|
|
self.fusedActivationFunction = subOptions.FusedActivationFunction()
|
|
self.potScaleInt16 = subOptions.PotScaleInt16()
|
|
|
|
# SubOptionsT
|
|
def Pack(self, builder):
|
|
SubOptionsStart(builder)
|
|
SubOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
SubOptionsAddPotScaleInt16(builder, self.potScaleInt16)
|
|
subOptions = SubOptionsEnd(builder)
|
|
return subOptions
|
|
|
|
|
|
class DivOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DivOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDivOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DivOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DivOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# DivOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def DivOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def DivOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(0, fusedActivationFunction, 0)
|
|
|
|
def DivOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DivOptionsT(object):
|
|
|
|
# DivOptionsT
|
|
def __init__(self):
|
|
self.fusedActivationFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
divOptions = DivOptions()
|
|
divOptions.Init(buf, pos)
|
|
return cls.InitFromObj(divOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, divOptions):
|
|
x = DivOptionsT()
|
|
x._UnPack(divOptions)
|
|
return x
|
|
|
|
# DivOptionsT
|
|
def _UnPack(self, divOptions):
|
|
if divOptions is None:
|
|
return
|
|
self.fusedActivationFunction = divOptions.FusedActivationFunction()
|
|
|
|
# DivOptionsT
|
|
def Pack(self, builder):
|
|
DivOptionsStart(builder)
|
|
DivOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
divOptions = DivOptionsEnd(builder)
|
|
return divOptions
|
|
|
|
|
|
class TopKV2Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = TopKV2Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTopKV2Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TopKV2OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# TopKV2Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def TopKV2OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def TopKV2OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class TopKV2OptionsT(object):
|
|
|
|
# TopKV2OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
topKv2Options = TopKV2Options()
|
|
topKv2Options.Init(buf, pos)
|
|
return cls.InitFromObj(topKv2Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, topKv2Options):
|
|
x = TopKV2OptionsT()
|
|
x._UnPack(topKv2Options)
|
|
return x
|
|
|
|
# TopKV2OptionsT
|
|
def _UnPack(self, topKv2Options):
|
|
if topKv2Options is None:
|
|
return
|
|
|
|
# TopKV2OptionsT
|
|
def Pack(self, builder):
|
|
TopKV2OptionsStart(builder)
|
|
topKv2Options = TopKV2OptionsEnd(builder)
|
|
return topKv2Options
|
|
|
|
|
|
class EmbeddingLookupSparseOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = EmbeddingLookupSparseOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsEmbeddingLookupSparseOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def EmbeddingLookupSparseOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# EmbeddingLookupSparseOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# EmbeddingLookupSparseOptions
|
|
def Combiner(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def EmbeddingLookupSparseOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def EmbeddingLookupSparseOptionsAddCombiner(builder, combiner):
|
|
builder.PrependInt8Slot(0, combiner, 0)
|
|
|
|
def EmbeddingLookupSparseOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class EmbeddingLookupSparseOptionsT(object):
|
|
|
|
# EmbeddingLookupSparseOptionsT
|
|
def __init__(self):
|
|
self.combiner = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
embeddingLookupSparseOptions = EmbeddingLookupSparseOptions()
|
|
embeddingLookupSparseOptions.Init(buf, pos)
|
|
return cls.InitFromObj(embeddingLookupSparseOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, embeddingLookupSparseOptions):
|
|
x = EmbeddingLookupSparseOptionsT()
|
|
x._UnPack(embeddingLookupSparseOptions)
|
|
return x
|
|
|
|
# EmbeddingLookupSparseOptionsT
|
|
def _UnPack(self, embeddingLookupSparseOptions):
|
|
if embeddingLookupSparseOptions is None:
|
|
return
|
|
self.combiner = embeddingLookupSparseOptions.Combiner()
|
|
|
|
# EmbeddingLookupSparseOptionsT
|
|
def Pack(self, builder):
|
|
EmbeddingLookupSparseOptionsStart(builder)
|
|
EmbeddingLookupSparseOptionsAddCombiner(builder, self.combiner)
|
|
embeddingLookupSparseOptions = EmbeddingLookupSparseOptionsEnd(builder)
|
|
return embeddingLookupSparseOptions
|
|
|
|
|
|
class GatherOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = GatherOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsGatherOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def GatherOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# GatherOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# GatherOptions
|
|
def Axis(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# GatherOptions
|
|
def BatchDims(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def GatherOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def GatherOptionsAddAxis(builder, axis):
|
|
builder.PrependInt32Slot(0, axis, 0)
|
|
|
|
def GatherOptionsAddBatchDims(builder, batchDims):
|
|
builder.PrependInt32Slot(1, batchDims, 0)
|
|
|
|
def GatherOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class GatherOptionsT(object):
|
|
|
|
# GatherOptionsT
|
|
def __init__(self):
|
|
self.axis = 0 # type: int
|
|
self.batchDims = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
gatherOptions = GatherOptions()
|
|
gatherOptions.Init(buf, pos)
|
|
return cls.InitFromObj(gatherOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, gatherOptions):
|
|
x = GatherOptionsT()
|
|
x._UnPack(gatherOptions)
|
|
return x
|
|
|
|
# GatherOptionsT
|
|
def _UnPack(self, gatherOptions):
|
|
if gatherOptions is None:
|
|
return
|
|
self.axis = gatherOptions.Axis()
|
|
self.batchDims = gatherOptions.BatchDims()
|
|
|
|
# GatherOptionsT
|
|
def Pack(self, builder):
|
|
GatherOptionsStart(builder)
|
|
GatherOptionsAddAxis(builder, self.axis)
|
|
GatherOptionsAddBatchDims(builder, self.batchDims)
|
|
gatherOptions = GatherOptionsEnd(builder)
|
|
return gatherOptions
|
|
|
|
|
|
class TransposeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = TransposeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTransposeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TransposeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# TransposeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def TransposeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def TransposeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class TransposeOptionsT(object):
|
|
|
|
# TransposeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
transposeOptions = TransposeOptions()
|
|
transposeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(transposeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, transposeOptions):
|
|
x = TransposeOptionsT()
|
|
x._UnPack(transposeOptions)
|
|
return x
|
|
|
|
# TransposeOptionsT
|
|
def _UnPack(self, transposeOptions):
|
|
if transposeOptions is None:
|
|
return
|
|
|
|
# TransposeOptionsT
|
|
def Pack(self, builder):
|
|
TransposeOptionsStart(builder)
|
|
transposeOptions = TransposeOptionsEnd(builder)
|
|
return transposeOptions
|
|
|
|
|
|
class ExpOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ExpOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsExpOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ExpOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ExpOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ExpOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ExpOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ExpOptionsT(object):
|
|
|
|
# ExpOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
expOptions = ExpOptions()
|
|
expOptions.Init(buf, pos)
|
|
return cls.InitFromObj(expOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, expOptions):
|
|
x = ExpOptionsT()
|
|
x._UnPack(expOptions)
|
|
return x
|
|
|
|
# ExpOptionsT
|
|
def _UnPack(self, expOptions):
|
|
if expOptions is None:
|
|
return
|
|
|
|
# ExpOptionsT
|
|
def Pack(self, builder):
|
|
ExpOptionsStart(builder)
|
|
expOptions = ExpOptionsEnd(builder)
|
|
return expOptions
|
|
|
|
|
|
class CosOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CosOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCosOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CosOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CosOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def CosOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def CosOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class CosOptionsT(object):
|
|
|
|
# CosOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
cosOptions = CosOptions()
|
|
cosOptions.Init(buf, pos)
|
|
return cls.InitFromObj(cosOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, cosOptions):
|
|
x = CosOptionsT()
|
|
x._UnPack(cosOptions)
|
|
return x
|
|
|
|
# CosOptionsT
|
|
def _UnPack(self, cosOptions):
|
|
if cosOptions is None:
|
|
return
|
|
|
|
# CosOptionsT
|
|
def Pack(self, builder):
|
|
CosOptionsStart(builder)
|
|
cosOptions = CosOptionsEnd(builder)
|
|
return cosOptions
|
|
|
|
|
|
class ReducerOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReducerOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReducerOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReducerOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReducerOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ReducerOptions
|
|
def KeepDims(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def ReducerOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ReducerOptionsAddKeepDims(builder, keepDims):
|
|
builder.PrependBoolSlot(0, keepDims, 0)
|
|
|
|
def ReducerOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ReducerOptionsT(object):
|
|
|
|
# ReducerOptionsT
|
|
def __init__(self):
|
|
self.keepDims = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
reducerOptions = ReducerOptions()
|
|
reducerOptions.Init(buf, pos)
|
|
return cls.InitFromObj(reducerOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, reducerOptions):
|
|
x = ReducerOptionsT()
|
|
x._UnPack(reducerOptions)
|
|
return x
|
|
|
|
# ReducerOptionsT
|
|
def _UnPack(self, reducerOptions):
|
|
if reducerOptions is None:
|
|
return
|
|
self.keepDims = reducerOptions.KeepDims()
|
|
|
|
# ReducerOptionsT
|
|
def Pack(self, builder):
|
|
ReducerOptionsStart(builder)
|
|
ReducerOptionsAddKeepDims(builder, self.keepDims)
|
|
reducerOptions = ReducerOptionsEnd(builder)
|
|
return reducerOptions
|
|
|
|
|
|
class SqueezeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SqueezeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSqueezeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SqueezeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SqueezeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SqueezeOptions
|
|
def SqueezeDims(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# SqueezeOptions
|
|
def SqueezeDimsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# SqueezeOptions
|
|
def SqueezeDimsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SqueezeOptions
|
|
def SqueezeDimsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def SqueezeOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SqueezeOptionsAddSqueezeDims(builder, squeezeDims):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(squeezeDims), 0)
|
|
|
|
def SqueezeOptionsStartSqueezeDimsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SqueezeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class SqueezeOptionsT(object):
|
|
|
|
# SqueezeOptionsT
|
|
def __init__(self):
|
|
self.squeezeDims = None # type: List[int]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
squeezeOptions = SqueezeOptions()
|
|
squeezeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(squeezeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, squeezeOptions):
|
|
x = SqueezeOptionsT()
|
|
x._UnPack(squeezeOptions)
|
|
return x
|
|
|
|
# SqueezeOptionsT
|
|
def _UnPack(self, squeezeOptions):
|
|
if squeezeOptions is None:
|
|
return
|
|
if not squeezeOptions.SqueezeDimsIsNone():
|
|
if np is None:
|
|
self.squeezeDims = []
|
|
for i in range(squeezeOptions.SqueezeDimsLength()):
|
|
self.squeezeDims.append(squeezeOptions.SqueezeDims(i))
|
|
else:
|
|
self.squeezeDims = squeezeOptions.SqueezeDimsAsNumpy()
|
|
|
|
# SqueezeOptionsT
|
|
def Pack(self, builder):
|
|
if self.squeezeDims is not None:
|
|
if np is not None and type(self.squeezeDims) is np.ndarray:
|
|
squeezeDims = builder.CreateNumpyVector(self.squeezeDims)
|
|
else:
|
|
SqueezeOptionsStartSqueezeDimsVector(builder, len(self.squeezeDims))
|
|
for i in reversed(range(len(self.squeezeDims))):
|
|
builder.PrependInt32(self.squeezeDims[i])
|
|
squeezeDims = builder.EndVector()
|
|
SqueezeOptionsStart(builder)
|
|
if self.squeezeDims is not None:
|
|
SqueezeOptionsAddSqueezeDims(builder, squeezeDims)
|
|
squeezeOptions = SqueezeOptionsEnd(builder)
|
|
return squeezeOptions
|
|
|
|
|
|
class SplitOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SplitOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSplitOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SplitOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SplitOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SplitOptions
|
|
def NumSplits(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def SplitOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SplitOptionsAddNumSplits(builder, numSplits):
|
|
builder.PrependInt32Slot(0, numSplits, 0)
|
|
|
|
def SplitOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SplitOptionsT(object):
|
|
|
|
# SplitOptionsT
|
|
def __init__(self):
|
|
self.numSplits = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
splitOptions = SplitOptions()
|
|
splitOptions.Init(buf, pos)
|
|
return cls.InitFromObj(splitOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, splitOptions):
|
|
x = SplitOptionsT()
|
|
x._UnPack(splitOptions)
|
|
return x
|
|
|
|
# SplitOptionsT
|
|
def _UnPack(self, splitOptions):
|
|
if splitOptions is None:
|
|
return
|
|
self.numSplits = splitOptions.NumSplits()
|
|
|
|
# SplitOptionsT
|
|
def Pack(self, builder):
|
|
SplitOptionsStart(builder)
|
|
SplitOptionsAddNumSplits(builder, self.numSplits)
|
|
splitOptions = SplitOptionsEnd(builder)
|
|
return splitOptions
|
|
|
|
|
|
class SplitVOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SplitVOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSplitVOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SplitVOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SplitVOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SplitVOptions
|
|
def NumSplits(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def SplitVOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SplitVOptionsAddNumSplits(builder, numSplits):
|
|
builder.PrependInt32Slot(0, numSplits, 0)
|
|
|
|
def SplitVOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SplitVOptionsT(object):
|
|
|
|
# SplitVOptionsT
|
|
def __init__(self):
|
|
self.numSplits = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
splitVoptions = SplitVOptions()
|
|
splitVoptions.Init(buf, pos)
|
|
return cls.InitFromObj(splitVoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, splitVoptions):
|
|
x = SplitVOptionsT()
|
|
x._UnPack(splitVoptions)
|
|
return x
|
|
|
|
# SplitVOptionsT
|
|
def _UnPack(self, splitVoptions):
|
|
if splitVoptions is None:
|
|
return
|
|
self.numSplits = splitVoptions.NumSplits()
|
|
|
|
# SplitVOptionsT
|
|
def Pack(self, builder):
|
|
SplitVOptionsStart(builder)
|
|
SplitVOptionsAddNumSplits(builder, self.numSplits)
|
|
splitVoptions = SplitVOptionsEnd(builder)
|
|
return splitVoptions
|
|
|
|
|
|
class StridedSliceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = StridedSliceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsStridedSliceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def StridedSliceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# StridedSliceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# StridedSliceOptions
|
|
def BeginMask(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StridedSliceOptions
|
|
def EndMask(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StridedSliceOptions
|
|
def EllipsisMask(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StridedSliceOptions
|
|
def NewAxisMask(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StridedSliceOptions
|
|
def ShrinkAxisMask(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# StridedSliceOptions
|
|
def Offset(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def StridedSliceOptionsStart(builder):
|
|
builder.StartObject(6)
|
|
|
|
def StridedSliceOptionsAddBeginMask(builder, beginMask):
|
|
builder.PrependInt32Slot(0, beginMask, 0)
|
|
|
|
def StridedSliceOptionsAddEndMask(builder, endMask):
|
|
builder.PrependInt32Slot(1, endMask, 0)
|
|
|
|
def StridedSliceOptionsAddEllipsisMask(builder, ellipsisMask):
|
|
builder.PrependInt32Slot(2, ellipsisMask, 0)
|
|
|
|
def StridedSliceOptionsAddNewAxisMask(builder, newAxisMask):
|
|
builder.PrependInt32Slot(3, newAxisMask, 0)
|
|
|
|
def StridedSliceOptionsAddShrinkAxisMask(builder, shrinkAxisMask):
|
|
builder.PrependInt32Slot(4, shrinkAxisMask, 0)
|
|
|
|
def StridedSliceOptionsAddOffset(builder, offset):
|
|
builder.PrependBoolSlot(5, offset, 0)
|
|
|
|
def StridedSliceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class StridedSliceOptionsT(object):
|
|
|
|
# StridedSliceOptionsT
|
|
def __init__(self):
|
|
self.beginMask = 0 # type: int
|
|
self.endMask = 0 # type: int
|
|
self.ellipsisMask = 0 # type: int
|
|
self.newAxisMask = 0 # type: int
|
|
self.shrinkAxisMask = 0 # type: int
|
|
self.offset = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
stridedSliceOptions = StridedSliceOptions()
|
|
stridedSliceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(stridedSliceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, stridedSliceOptions):
|
|
x = StridedSliceOptionsT()
|
|
x._UnPack(stridedSliceOptions)
|
|
return x
|
|
|
|
# StridedSliceOptionsT
|
|
def _UnPack(self, stridedSliceOptions):
|
|
if stridedSliceOptions is None:
|
|
return
|
|
self.beginMask = stridedSliceOptions.BeginMask()
|
|
self.endMask = stridedSliceOptions.EndMask()
|
|
self.ellipsisMask = stridedSliceOptions.EllipsisMask()
|
|
self.newAxisMask = stridedSliceOptions.NewAxisMask()
|
|
self.shrinkAxisMask = stridedSliceOptions.ShrinkAxisMask()
|
|
self.offset = stridedSliceOptions.Offset()
|
|
|
|
# StridedSliceOptionsT
|
|
def Pack(self, builder):
|
|
StridedSliceOptionsStart(builder)
|
|
StridedSliceOptionsAddBeginMask(builder, self.beginMask)
|
|
StridedSliceOptionsAddEndMask(builder, self.endMask)
|
|
StridedSliceOptionsAddEllipsisMask(builder, self.ellipsisMask)
|
|
StridedSliceOptionsAddNewAxisMask(builder, self.newAxisMask)
|
|
StridedSliceOptionsAddShrinkAxisMask(builder, self.shrinkAxisMask)
|
|
StridedSliceOptionsAddOffset(builder, self.offset)
|
|
stridedSliceOptions = StridedSliceOptionsEnd(builder)
|
|
return stridedSliceOptions
|
|
|
|
|
|
class LogSoftmaxOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LogSoftmaxOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLogSoftmaxOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LogSoftmaxOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LogSoftmaxOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LogSoftmaxOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LogSoftmaxOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LogSoftmaxOptionsT(object):
|
|
|
|
# LogSoftmaxOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
logSoftmaxOptions = LogSoftmaxOptions()
|
|
logSoftmaxOptions.Init(buf, pos)
|
|
return cls.InitFromObj(logSoftmaxOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, logSoftmaxOptions):
|
|
x = LogSoftmaxOptionsT()
|
|
x._UnPack(logSoftmaxOptions)
|
|
return x
|
|
|
|
# LogSoftmaxOptionsT
|
|
def _UnPack(self, logSoftmaxOptions):
|
|
if logSoftmaxOptions is None:
|
|
return
|
|
|
|
# LogSoftmaxOptionsT
|
|
def Pack(self, builder):
|
|
LogSoftmaxOptionsStart(builder)
|
|
logSoftmaxOptions = LogSoftmaxOptionsEnd(builder)
|
|
return logSoftmaxOptions
|
|
|
|
|
|
class CastOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CastOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCastOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CastOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CastOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# CastOptions
|
|
def InDataType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# CastOptions
|
|
def OutDataType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def CastOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def CastOptionsAddInDataType(builder, inDataType):
|
|
builder.PrependInt8Slot(0, inDataType, 0)
|
|
|
|
def CastOptionsAddOutDataType(builder, outDataType):
|
|
builder.PrependInt8Slot(1, outDataType, 0)
|
|
|
|
def CastOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class CastOptionsT(object):
|
|
|
|
# CastOptionsT
|
|
def __init__(self):
|
|
self.inDataType = 0 # type: int
|
|
self.outDataType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
castOptions = CastOptions()
|
|
castOptions.Init(buf, pos)
|
|
return cls.InitFromObj(castOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, castOptions):
|
|
x = CastOptionsT()
|
|
x._UnPack(castOptions)
|
|
return x
|
|
|
|
# CastOptionsT
|
|
def _UnPack(self, castOptions):
|
|
if castOptions is None:
|
|
return
|
|
self.inDataType = castOptions.InDataType()
|
|
self.outDataType = castOptions.OutDataType()
|
|
|
|
# CastOptionsT
|
|
def Pack(self, builder):
|
|
CastOptionsStart(builder)
|
|
CastOptionsAddInDataType(builder, self.inDataType)
|
|
CastOptionsAddOutDataType(builder, self.outDataType)
|
|
castOptions = CastOptionsEnd(builder)
|
|
return castOptions
|
|
|
|
|
|
class DequantizeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DequantizeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDequantizeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DequantizeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DequantizeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def DequantizeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def DequantizeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DequantizeOptionsT(object):
|
|
|
|
# DequantizeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
dequantizeOptions = DequantizeOptions()
|
|
dequantizeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(dequantizeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, dequantizeOptions):
|
|
x = DequantizeOptionsT()
|
|
x._UnPack(dequantizeOptions)
|
|
return x
|
|
|
|
# DequantizeOptionsT
|
|
def _UnPack(self, dequantizeOptions):
|
|
if dequantizeOptions is None:
|
|
return
|
|
|
|
# DequantizeOptionsT
|
|
def Pack(self, builder):
|
|
DequantizeOptionsStart(builder)
|
|
dequantizeOptions = DequantizeOptionsEnd(builder)
|
|
return dequantizeOptions
|
|
|
|
|
|
class MaximumMinimumOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = MaximumMinimumOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMaximumMinimumOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MaximumMinimumOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# MaximumMinimumOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def MaximumMinimumOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def MaximumMinimumOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MaximumMinimumOptionsT(object):
|
|
|
|
# MaximumMinimumOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
maximumMinimumOptions = MaximumMinimumOptions()
|
|
maximumMinimumOptions.Init(buf, pos)
|
|
return cls.InitFromObj(maximumMinimumOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, maximumMinimumOptions):
|
|
x = MaximumMinimumOptionsT()
|
|
x._UnPack(maximumMinimumOptions)
|
|
return x
|
|
|
|
# MaximumMinimumOptionsT
|
|
def _UnPack(self, maximumMinimumOptions):
|
|
if maximumMinimumOptions is None:
|
|
return
|
|
|
|
# MaximumMinimumOptionsT
|
|
def Pack(self, builder):
|
|
MaximumMinimumOptionsStart(builder)
|
|
maximumMinimumOptions = MaximumMinimumOptionsEnd(builder)
|
|
return maximumMinimumOptions
|
|
|
|
|
|
class TileOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = TileOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTileOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TileOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# TileOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def TileOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def TileOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class TileOptionsT(object):
|
|
|
|
# TileOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
tileOptions = TileOptions()
|
|
tileOptions.Init(buf, pos)
|
|
return cls.InitFromObj(tileOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, tileOptions):
|
|
x = TileOptionsT()
|
|
x._UnPack(tileOptions)
|
|
return x
|
|
|
|
# TileOptionsT
|
|
def _UnPack(self, tileOptions):
|
|
if tileOptions is None:
|
|
return
|
|
|
|
# TileOptionsT
|
|
def Pack(self, builder):
|
|
TileOptionsStart(builder)
|
|
tileOptions = TileOptionsEnd(builder)
|
|
return tileOptions
|
|
|
|
|
|
class ArgMaxOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ArgMaxOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsArgMaxOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ArgMaxOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ArgMaxOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ArgMaxOptions
|
|
def OutputType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ArgMaxOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ArgMaxOptionsAddOutputType(builder, outputType):
|
|
builder.PrependInt8Slot(0, outputType, 0)
|
|
|
|
def ArgMaxOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ArgMaxOptionsT(object):
|
|
|
|
# ArgMaxOptionsT
|
|
def __init__(self):
|
|
self.outputType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
argMaxOptions = ArgMaxOptions()
|
|
argMaxOptions.Init(buf, pos)
|
|
return cls.InitFromObj(argMaxOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, argMaxOptions):
|
|
x = ArgMaxOptionsT()
|
|
x._UnPack(argMaxOptions)
|
|
return x
|
|
|
|
# ArgMaxOptionsT
|
|
def _UnPack(self, argMaxOptions):
|
|
if argMaxOptions is None:
|
|
return
|
|
self.outputType = argMaxOptions.OutputType()
|
|
|
|
# ArgMaxOptionsT
|
|
def Pack(self, builder):
|
|
ArgMaxOptionsStart(builder)
|
|
ArgMaxOptionsAddOutputType(builder, self.outputType)
|
|
argMaxOptions = ArgMaxOptionsEnd(builder)
|
|
return argMaxOptions
|
|
|
|
|
|
class ArgMinOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ArgMinOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsArgMinOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ArgMinOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ArgMinOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ArgMinOptions
|
|
def OutputType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ArgMinOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ArgMinOptionsAddOutputType(builder, outputType):
|
|
builder.PrependInt8Slot(0, outputType, 0)
|
|
|
|
def ArgMinOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ArgMinOptionsT(object):
|
|
|
|
# ArgMinOptionsT
|
|
def __init__(self):
|
|
self.outputType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
argMinOptions = ArgMinOptions()
|
|
argMinOptions.Init(buf, pos)
|
|
return cls.InitFromObj(argMinOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, argMinOptions):
|
|
x = ArgMinOptionsT()
|
|
x._UnPack(argMinOptions)
|
|
return x
|
|
|
|
# ArgMinOptionsT
|
|
def _UnPack(self, argMinOptions):
|
|
if argMinOptions is None:
|
|
return
|
|
self.outputType = argMinOptions.OutputType()
|
|
|
|
# ArgMinOptionsT
|
|
def Pack(self, builder):
|
|
ArgMinOptionsStart(builder)
|
|
ArgMinOptionsAddOutputType(builder, self.outputType)
|
|
argMinOptions = ArgMinOptionsEnd(builder)
|
|
return argMinOptions
|
|
|
|
|
|
class GreaterOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = GreaterOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsGreaterOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def GreaterOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# GreaterOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def GreaterOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def GreaterOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class GreaterOptionsT(object):
|
|
|
|
# GreaterOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
greaterOptions = GreaterOptions()
|
|
greaterOptions.Init(buf, pos)
|
|
return cls.InitFromObj(greaterOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, greaterOptions):
|
|
x = GreaterOptionsT()
|
|
x._UnPack(greaterOptions)
|
|
return x
|
|
|
|
# GreaterOptionsT
|
|
def _UnPack(self, greaterOptions):
|
|
if greaterOptions is None:
|
|
return
|
|
|
|
# GreaterOptionsT
|
|
def Pack(self, builder):
|
|
GreaterOptionsStart(builder)
|
|
greaterOptions = GreaterOptionsEnd(builder)
|
|
return greaterOptions
|
|
|
|
|
|
class GreaterEqualOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = GreaterEqualOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsGreaterEqualOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def GreaterEqualOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# GreaterEqualOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def GreaterEqualOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def GreaterEqualOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class GreaterEqualOptionsT(object):
|
|
|
|
# GreaterEqualOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
greaterEqualOptions = GreaterEqualOptions()
|
|
greaterEqualOptions.Init(buf, pos)
|
|
return cls.InitFromObj(greaterEqualOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, greaterEqualOptions):
|
|
x = GreaterEqualOptionsT()
|
|
x._UnPack(greaterEqualOptions)
|
|
return x
|
|
|
|
# GreaterEqualOptionsT
|
|
def _UnPack(self, greaterEqualOptions):
|
|
if greaterEqualOptions is None:
|
|
return
|
|
|
|
# GreaterEqualOptionsT
|
|
def Pack(self, builder):
|
|
GreaterEqualOptionsStart(builder)
|
|
greaterEqualOptions = GreaterEqualOptionsEnd(builder)
|
|
return greaterEqualOptions
|
|
|
|
|
|
class LessOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LessOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLessOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LessOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LessOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LessOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LessOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LessOptionsT(object):
|
|
|
|
# LessOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
lessOptions = LessOptions()
|
|
lessOptions.Init(buf, pos)
|
|
return cls.InitFromObj(lessOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, lessOptions):
|
|
x = LessOptionsT()
|
|
x._UnPack(lessOptions)
|
|
return x
|
|
|
|
# LessOptionsT
|
|
def _UnPack(self, lessOptions):
|
|
if lessOptions is None:
|
|
return
|
|
|
|
# LessOptionsT
|
|
def Pack(self, builder):
|
|
LessOptionsStart(builder)
|
|
lessOptions = LessOptionsEnd(builder)
|
|
return lessOptions
|
|
|
|
|
|
class LessEqualOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LessEqualOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLessEqualOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LessEqualOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LessEqualOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LessEqualOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LessEqualOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LessEqualOptionsT(object):
|
|
|
|
# LessEqualOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
lessEqualOptions = LessEqualOptions()
|
|
lessEqualOptions.Init(buf, pos)
|
|
return cls.InitFromObj(lessEqualOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, lessEqualOptions):
|
|
x = LessEqualOptionsT()
|
|
x._UnPack(lessEqualOptions)
|
|
return x
|
|
|
|
# LessEqualOptionsT
|
|
def _UnPack(self, lessEqualOptions):
|
|
if lessEqualOptions is None:
|
|
return
|
|
|
|
# LessEqualOptionsT
|
|
def Pack(self, builder):
|
|
LessEqualOptionsStart(builder)
|
|
lessEqualOptions = LessEqualOptionsEnd(builder)
|
|
return lessEqualOptions
|
|
|
|
|
|
class NegOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = NegOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsNegOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def NegOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# NegOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def NegOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def NegOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class NegOptionsT(object):
|
|
|
|
# NegOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
negOptions = NegOptions()
|
|
negOptions.Init(buf, pos)
|
|
return cls.InitFromObj(negOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, negOptions):
|
|
x = NegOptionsT()
|
|
x._UnPack(negOptions)
|
|
return x
|
|
|
|
# NegOptionsT
|
|
def _UnPack(self, negOptions):
|
|
if negOptions is None:
|
|
return
|
|
|
|
# NegOptionsT
|
|
def Pack(self, builder):
|
|
NegOptionsStart(builder)
|
|
negOptions = NegOptionsEnd(builder)
|
|
return negOptions
|
|
|
|
|
|
class SelectOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SelectOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSelectOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SelectOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SelectOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SelectOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SelectOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SelectOptionsT(object):
|
|
|
|
# SelectOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
selectOptions = SelectOptions()
|
|
selectOptions.Init(buf, pos)
|
|
return cls.InitFromObj(selectOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, selectOptions):
|
|
x = SelectOptionsT()
|
|
x._UnPack(selectOptions)
|
|
return x
|
|
|
|
# SelectOptionsT
|
|
def _UnPack(self, selectOptions):
|
|
if selectOptions is None:
|
|
return
|
|
|
|
# SelectOptionsT
|
|
def Pack(self, builder):
|
|
SelectOptionsStart(builder)
|
|
selectOptions = SelectOptionsEnd(builder)
|
|
return selectOptions
|
|
|
|
|
|
class SliceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SliceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSliceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SliceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SliceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SliceOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SliceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SliceOptionsT(object):
|
|
|
|
# SliceOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
sliceOptions = SliceOptions()
|
|
sliceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(sliceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, sliceOptions):
|
|
x = SliceOptionsT()
|
|
x._UnPack(sliceOptions)
|
|
return x
|
|
|
|
# SliceOptionsT
|
|
def _UnPack(self, sliceOptions):
|
|
if sliceOptions is None:
|
|
return
|
|
|
|
# SliceOptionsT
|
|
def Pack(self, builder):
|
|
SliceOptionsStart(builder)
|
|
sliceOptions = SliceOptionsEnd(builder)
|
|
return sliceOptions
|
|
|
|
|
|
class TransposeConvOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = TransposeConvOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTransposeConvOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TransposeConvOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# TransposeConvOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# TransposeConvOptions
|
|
def Padding(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# TransposeConvOptions
|
|
def StrideW(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# TransposeConvOptions
|
|
def StrideH(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# TransposeConvOptions
|
|
def FusedActivationFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# TransposeConvOptions
|
|
def QuantizedBiasType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def TransposeConvOptionsStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def TransposeConvOptionsAddPadding(builder, padding):
|
|
builder.PrependInt8Slot(0, padding, 0)
|
|
|
|
def TransposeConvOptionsAddStrideW(builder, strideW):
|
|
builder.PrependInt32Slot(1, strideW, 0)
|
|
|
|
def TransposeConvOptionsAddStrideH(builder, strideH):
|
|
builder.PrependInt32Slot(2, strideH, 0)
|
|
|
|
def TransposeConvOptionsAddFusedActivationFunction(builder, fusedActivationFunction):
|
|
builder.PrependInt8Slot(3, fusedActivationFunction, 0)
|
|
|
|
def TransposeConvOptionsAddQuantizedBiasType(builder, quantizedBiasType):
|
|
builder.PrependInt8Slot(4, quantizedBiasType, 0)
|
|
|
|
def TransposeConvOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class TransposeConvOptionsT(object):
|
|
|
|
# TransposeConvOptionsT
|
|
def __init__(self):
|
|
self.padding = 0 # type: int
|
|
self.strideW = 0 # type: int
|
|
self.strideH = 0 # type: int
|
|
self.fusedActivationFunction = 0 # type: int
|
|
self.quantizedBiasType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
transposeConvOptions = TransposeConvOptions()
|
|
transposeConvOptions.Init(buf, pos)
|
|
return cls.InitFromObj(transposeConvOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, transposeConvOptions):
|
|
x = TransposeConvOptionsT()
|
|
x._UnPack(transposeConvOptions)
|
|
return x
|
|
|
|
# TransposeConvOptionsT
|
|
def _UnPack(self, transposeConvOptions):
|
|
if transposeConvOptions is None:
|
|
return
|
|
self.padding = transposeConvOptions.Padding()
|
|
self.strideW = transposeConvOptions.StrideW()
|
|
self.strideH = transposeConvOptions.StrideH()
|
|
self.fusedActivationFunction = transposeConvOptions.FusedActivationFunction()
|
|
self.quantizedBiasType = transposeConvOptions.QuantizedBiasType()
|
|
|
|
# TransposeConvOptionsT
|
|
def Pack(self, builder):
|
|
TransposeConvOptionsStart(builder)
|
|
TransposeConvOptionsAddPadding(builder, self.padding)
|
|
TransposeConvOptionsAddStrideW(builder, self.strideW)
|
|
TransposeConvOptionsAddStrideH(builder, self.strideH)
|
|
TransposeConvOptionsAddFusedActivationFunction(builder, self.fusedActivationFunction)
|
|
TransposeConvOptionsAddQuantizedBiasType(builder, self.quantizedBiasType)
|
|
transposeConvOptions = TransposeConvOptionsEnd(builder)
|
|
return transposeConvOptions
|
|
|
|
|
|
class ExpandDimsOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ExpandDimsOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsExpandDimsOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ExpandDimsOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ExpandDimsOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ExpandDimsOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ExpandDimsOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ExpandDimsOptionsT(object):
|
|
|
|
# ExpandDimsOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
expandDimsOptions = ExpandDimsOptions()
|
|
expandDimsOptions.Init(buf, pos)
|
|
return cls.InitFromObj(expandDimsOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, expandDimsOptions):
|
|
x = ExpandDimsOptionsT()
|
|
x._UnPack(expandDimsOptions)
|
|
return x
|
|
|
|
# ExpandDimsOptionsT
|
|
def _UnPack(self, expandDimsOptions):
|
|
if expandDimsOptions is None:
|
|
return
|
|
|
|
# ExpandDimsOptionsT
|
|
def Pack(self, builder):
|
|
ExpandDimsOptionsStart(builder)
|
|
expandDimsOptions = ExpandDimsOptionsEnd(builder)
|
|
return expandDimsOptions
|
|
|
|
|
|
class SparseToDenseOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SparseToDenseOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSparseToDenseOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SparseToDenseOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SparseToDenseOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SparseToDenseOptions
|
|
def ValidateIndices(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def SparseToDenseOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def SparseToDenseOptionsAddValidateIndices(builder, validateIndices):
|
|
builder.PrependBoolSlot(0, validateIndices, 0)
|
|
|
|
def SparseToDenseOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SparseToDenseOptionsT(object):
|
|
|
|
# SparseToDenseOptionsT
|
|
def __init__(self):
|
|
self.validateIndices = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
sparseToDenseOptions = SparseToDenseOptions()
|
|
sparseToDenseOptions.Init(buf, pos)
|
|
return cls.InitFromObj(sparseToDenseOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, sparseToDenseOptions):
|
|
x = SparseToDenseOptionsT()
|
|
x._UnPack(sparseToDenseOptions)
|
|
return x
|
|
|
|
# SparseToDenseOptionsT
|
|
def _UnPack(self, sparseToDenseOptions):
|
|
if sparseToDenseOptions is None:
|
|
return
|
|
self.validateIndices = sparseToDenseOptions.ValidateIndices()
|
|
|
|
# SparseToDenseOptionsT
|
|
def Pack(self, builder):
|
|
SparseToDenseOptionsStart(builder)
|
|
SparseToDenseOptionsAddValidateIndices(builder, self.validateIndices)
|
|
sparseToDenseOptions = SparseToDenseOptionsEnd(builder)
|
|
return sparseToDenseOptions
|
|
|
|
|
|
class EqualOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = EqualOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsEqualOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def EqualOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# EqualOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def EqualOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def EqualOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class EqualOptionsT(object):
|
|
|
|
# EqualOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
equalOptions = EqualOptions()
|
|
equalOptions.Init(buf, pos)
|
|
return cls.InitFromObj(equalOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, equalOptions):
|
|
x = EqualOptionsT()
|
|
x._UnPack(equalOptions)
|
|
return x
|
|
|
|
# EqualOptionsT
|
|
def _UnPack(self, equalOptions):
|
|
if equalOptions is None:
|
|
return
|
|
|
|
# EqualOptionsT
|
|
def Pack(self, builder):
|
|
EqualOptionsStart(builder)
|
|
equalOptions = EqualOptionsEnd(builder)
|
|
return equalOptions
|
|
|
|
|
|
class NotEqualOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = NotEqualOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsNotEqualOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def NotEqualOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# NotEqualOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def NotEqualOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def NotEqualOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class NotEqualOptionsT(object):
|
|
|
|
# NotEqualOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
notEqualOptions = NotEqualOptions()
|
|
notEqualOptions.Init(buf, pos)
|
|
return cls.InitFromObj(notEqualOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, notEqualOptions):
|
|
x = NotEqualOptionsT()
|
|
x._UnPack(notEqualOptions)
|
|
return x
|
|
|
|
# NotEqualOptionsT
|
|
def _UnPack(self, notEqualOptions):
|
|
if notEqualOptions is None:
|
|
return
|
|
|
|
# NotEqualOptionsT
|
|
def Pack(self, builder):
|
|
NotEqualOptionsStart(builder)
|
|
notEqualOptions = NotEqualOptionsEnd(builder)
|
|
return notEqualOptions
|
|
|
|
|
|
class ShapeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ShapeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsShapeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ShapeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ShapeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ShapeOptions
|
|
def OutType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ShapeOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ShapeOptionsAddOutType(builder, outType):
|
|
builder.PrependInt8Slot(0, outType, 0)
|
|
|
|
def ShapeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ShapeOptionsT(object):
|
|
|
|
# ShapeOptionsT
|
|
def __init__(self):
|
|
self.outType = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
shapeOptions = ShapeOptions()
|
|
shapeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(shapeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, shapeOptions):
|
|
x = ShapeOptionsT()
|
|
x._UnPack(shapeOptions)
|
|
return x
|
|
|
|
# ShapeOptionsT
|
|
def _UnPack(self, shapeOptions):
|
|
if shapeOptions is None:
|
|
return
|
|
self.outType = shapeOptions.OutType()
|
|
|
|
# ShapeOptionsT
|
|
def Pack(self, builder):
|
|
ShapeOptionsStart(builder)
|
|
ShapeOptionsAddOutType(builder, self.outType)
|
|
shapeOptions = ShapeOptionsEnd(builder)
|
|
return shapeOptions
|
|
|
|
|
|
class RankOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = RankOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRankOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def RankOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# RankOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def RankOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def RankOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class RankOptionsT(object):
|
|
|
|
# RankOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
rankOptions = RankOptions()
|
|
rankOptions.Init(buf, pos)
|
|
return cls.InitFromObj(rankOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, rankOptions):
|
|
x = RankOptionsT()
|
|
x._UnPack(rankOptions)
|
|
return x
|
|
|
|
# RankOptionsT
|
|
def _UnPack(self, rankOptions):
|
|
if rankOptions is None:
|
|
return
|
|
|
|
# RankOptionsT
|
|
def Pack(self, builder):
|
|
RankOptionsStart(builder)
|
|
rankOptions = RankOptionsEnd(builder)
|
|
return rankOptions
|
|
|
|
|
|
class PowOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = PowOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsPowOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def PowOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# PowOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def PowOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def PowOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class PowOptionsT(object):
|
|
|
|
# PowOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
powOptions = PowOptions()
|
|
powOptions.Init(buf, pos)
|
|
return cls.InitFromObj(powOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, powOptions):
|
|
x = PowOptionsT()
|
|
x._UnPack(powOptions)
|
|
return x
|
|
|
|
# PowOptionsT
|
|
def _UnPack(self, powOptions):
|
|
if powOptions is None:
|
|
return
|
|
|
|
# PowOptionsT
|
|
def Pack(self, builder):
|
|
PowOptionsStart(builder)
|
|
powOptions = PowOptionsEnd(builder)
|
|
return powOptions
|
|
|
|
|
|
class FakeQuantOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = FakeQuantOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsFakeQuantOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def FakeQuantOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# FakeQuantOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# FakeQuantOptions
|
|
def Min(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# FakeQuantOptions
|
|
def Max(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
# FakeQuantOptions
|
|
def NumBits(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# FakeQuantOptions
|
|
def NarrowRange(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def FakeQuantOptionsStart(builder):
|
|
builder.StartObject(4)
|
|
|
|
def FakeQuantOptionsAddMin(builder, min):
|
|
builder.PrependFloat32Slot(0, min, 0.0)
|
|
|
|
def FakeQuantOptionsAddMax(builder, max):
|
|
builder.PrependFloat32Slot(1, max, 0.0)
|
|
|
|
def FakeQuantOptionsAddNumBits(builder, numBits):
|
|
builder.PrependInt32Slot(2, numBits, 0)
|
|
|
|
def FakeQuantOptionsAddNarrowRange(builder, narrowRange):
|
|
builder.PrependBoolSlot(3, narrowRange, 0)
|
|
|
|
def FakeQuantOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class FakeQuantOptionsT(object):
|
|
|
|
# FakeQuantOptionsT
|
|
def __init__(self):
|
|
self.min = 0.0 # type: float
|
|
self.max = 0.0 # type: float
|
|
self.numBits = 0 # type: int
|
|
self.narrowRange = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
fakeQuantOptions = FakeQuantOptions()
|
|
fakeQuantOptions.Init(buf, pos)
|
|
return cls.InitFromObj(fakeQuantOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, fakeQuantOptions):
|
|
x = FakeQuantOptionsT()
|
|
x._UnPack(fakeQuantOptions)
|
|
return x
|
|
|
|
# FakeQuantOptionsT
|
|
def _UnPack(self, fakeQuantOptions):
|
|
if fakeQuantOptions is None:
|
|
return
|
|
self.min = fakeQuantOptions.Min()
|
|
self.max = fakeQuantOptions.Max()
|
|
self.numBits = fakeQuantOptions.NumBits()
|
|
self.narrowRange = fakeQuantOptions.NarrowRange()
|
|
|
|
# FakeQuantOptionsT
|
|
def Pack(self, builder):
|
|
FakeQuantOptionsStart(builder)
|
|
FakeQuantOptionsAddMin(builder, self.min)
|
|
FakeQuantOptionsAddMax(builder, self.max)
|
|
FakeQuantOptionsAddNumBits(builder, self.numBits)
|
|
FakeQuantOptionsAddNarrowRange(builder, self.narrowRange)
|
|
fakeQuantOptions = FakeQuantOptionsEnd(builder)
|
|
return fakeQuantOptions
|
|
|
|
|
|
class PackOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = PackOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsPackOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def PackOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# PackOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# PackOptions
|
|
def ValuesCount(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# PackOptions
|
|
def Axis(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def PackOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def PackOptionsAddValuesCount(builder, valuesCount):
|
|
builder.PrependInt32Slot(0, valuesCount, 0)
|
|
|
|
def PackOptionsAddAxis(builder, axis):
|
|
builder.PrependInt32Slot(1, axis, 0)
|
|
|
|
def PackOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class PackOptionsT(object):
|
|
|
|
# PackOptionsT
|
|
def __init__(self):
|
|
self.valuesCount = 0 # type: int
|
|
self.axis = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
packOptions = PackOptions()
|
|
packOptions.Init(buf, pos)
|
|
return cls.InitFromObj(packOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, packOptions):
|
|
x = PackOptionsT()
|
|
x._UnPack(packOptions)
|
|
return x
|
|
|
|
# PackOptionsT
|
|
def _UnPack(self, packOptions):
|
|
if packOptions is None:
|
|
return
|
|
self.valuesCount = packOptions.ValuesCount()
|
|
self.axis = packOptions.Axis()
|
|
|
|
# PackOptionsT
|
|
def Pack(self, builder):
|
|
PackOptionsStart(builder)
|
|
PackOptionsAddValuesCount(builder, self.valuesCount)
|
|
PackOptionsAddAxis(builder, self.axis)
|
|
packOptions = PackOptionsEnd(builder)
|
|
return packOptions
|
|
|
|
|
|
class LogicalOrOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LogicalOrOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLogicalOrOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LogicalOrOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LogicalOrOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LogicalOrOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LogicalOrOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LogicalOrOptionsT(object):
|
|
|
|
# LogicalOrOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
logicalOrOptions = LogicalOrOptions()
|
|
logicalOrOptions.Init(buf, pos)
|
|
return cls.InitFromObj(logicalOrOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, logicalOrOptions):
|
|
x = LogicalOrOptionsT()
|
|
x._UnPack(logicalOrOptions)
|
|
return x
|
|
|
|
# LogicalOrOptionsT
|
|
def _UnPack(self, logicalOrOptions):
|
|
if logicalOrOptions is None:
|
|
return
|
|
|
|
# LogicalOrOptionsT
|
|
def Pack(self, builder):
|
|
LogicalOrOptionsStart(builder)
|
|
logicalOrOptions = LogicalOrOptionsEnd(builder)
|
|
return logicalOrOptions
|
|
|
|
|
|
class OneHotOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = OneHotOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsOneHotOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def OneHotOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# OneHotOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# OneHotOptions
|
|
def Axis(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def OneHotOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def OneHotOptionsAddAxis(builder, axis):
|
|
builder.PrependInt32Slot(0, axis, 0)
|
|
|
|
def OneHotOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class OneHotOptionsT(object):
|
|
|
|
# OneHotOptionsT
|
|
def __init__(self):
|
|
self.axis = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
oneHotOptions = OneHotOptions()
|
|
oneHotOptions.Init(buf, pos)
|
|
return cls.InitFromObj(oneHotOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, oneHotOptions):
|
|
x = OneHotOptionsT()
|
|
x._UnPack(oneHotOptions)
|
|
return x
|
|
|
|
# OneHotOptionsT
|
|
def _UnPack(self, oneHotOptions):
|
|
if oneHotOptions is None:
|
|
return
|
|
self.axis = oneHotOptions.Axis()
|
|
|
|
# OneHotOptionsT
|
|
def Pack(self, builder):
|
|
OneHotOptionsStart(builder)
|
|
OneHotOptionsAddAxis(builder, self.axis)
|
|
oneHotOptions = OneHotOptionsEnd(builder)
|
|
return oneHotOptions
|
|
|
|
|
|
class AbsOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = AbsOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsAbsOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def AbsOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# AbsOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def AbsOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def AbsOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class AbsOptionsT(object):
|
|
|
|
# AbsOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
absOptions = AbsOptions()
|
|
absOptions.Init(buf, pos)
|
|
return cls.InitFromObj(absOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, absOptions):
|
|
x = AbsOptionsT()
|
|
x._UnPack(absOptions)
|
|
return x
|
|
|
|
# AbsOptionsT
|
|
def _UnPack(self, absOptions):
|
|
if absOptions is None:
|
|
return
|
|
|
|
# AbsOptionsT
|
|
def Pack(self, builder):
|
|
AbsOptionsStart(builder)
|
|
absOptions = AbsOptionsEnd(builder)
|
|
return absOptions
|
|
|
|
|
|
class HardSwishOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = HardSwishOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsHardSwishOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def HardSwishOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# HardSwishOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def HardSwishOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def HardSwishOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class HardSwishOptionsT(object):
|
|
|
|
# HardSwishOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
hardSwishOptions = HardSwishOptions()
|
|
hardSwishOptions.Init(buf, pos)
|
|
return cls.InitFromObj(hardSwishOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, hardSwishOptions):
|
|
x = HardSwishOptionsT()
|
|
x._UnPack(hardSwishOptions)
|
|
return x
|
|
|
|
# HardSwishOptionsT
|
|
def _UnPack(self, hardSwishOptions):
|
|
if hardSwishOptions is None:
|
|
return
|
|
|
|
# HardSwishOptionsT
|
|
def Pack(self, builder):
|
|
HardSwishOptionsStart(builder)
|
|
hardSwishOptions = HardSwishOptionsEnd(builder)
|
|
return hardSwishOptions
|
|
|
|
|
|
class LogicalAndOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LogicalAndOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLogicalAndOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LogicalAndOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LogicalAndOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LogicalAndOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LogicalAndOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LogicalAndOptionsT(object):
|
|
|
|
# LogicalAndOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
logicalAndOptions = LogicalAndOptions()
|
|
logicalAndOptions.Init(buf, pos)
|
|
return cls.InitFromObj(logicalAndOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, logicalAndOptions):
|
|
x = LogicalAndOptionsT()
|
|
x._UnPack(logicalAndOptions)
|
|
return x
|
|
|
|
# LogicalAndOptionsT
|
|
def _UnPack(self, logicalAndOptions):
|
|
if logicalAndOptions is None:
|
|
return
|
|
|
|
# LogicalAndOptionsT
|
|
def Pack(self, builder):
|
|
LogicalAndOptionsStart(builder)
|
|
logicalAndOptions = LogicalAndOptionsEnd(builder)
|
|
return logicalAndOptions
|
|
|
|
|
|
class LogicalNotOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LogicalNotOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLogicalNotOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LogicalNotOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LogicalNotOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def LogicalNotOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def LogicalNotOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LogicalNotOptionsT(object):
|
|
|
|
# LogicalNotOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
logicalNotOptions = LogicalNotOptions()
|
|
logicalNotOptions.Init(buf, pos)
|
|
return cls.InitFromObj(logicalNotOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, logicalNotOptions):
|
|
x = LogicalNotOptionsT()
|
|
x._UnPack(logicalNotOptions)
|
|
return x
|
|
|
|
# LogicalNotOptionsT
|
|
def _UnPack(self, logicalNotOptions):
|
|
if logicalNotOptions is None:
|
|
return
|
|
|
|
# LogicalNotOptionsT
|
|
def Pack(self, builder):
|
|
LogicalNotOptionsStart(builder)
|
|
logicalNotOptions = LogicalNotOptionsEnd(builder)
|
|
return logicalNotOptions
|
|
|
|
|
|
class UnpackOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnpackOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnpackOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnpackOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnpackOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# UnpackOptions
|
|
def Num(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# UnpackOptions
|
|
def Axis(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def UnpackOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def UnpackOptionsAddNum(builder, num):
|
|
builder.PrependInt32Slot(0, num, 0)
|
|
|
|
def UnpackOptionsAddAxis(builder, axis):
|
|
builder.PrependInt32Slot(1, axis, 0)
|
|
|
|
def UnpackOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnpackOptionsT(object):
|
|
|
|
# UnpackOptionsT
|
|
def __init__(self):
|
|
self.num = 0 # type: int
|
|
self.axis = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unpackOptions = UnpackOptions()
|
|
unpackOptions.Init(buf, pos)
|
|
return cls.InitFromObj(unpackOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unpackOptions):
|
|
x = UnpackOptionsT()
|
|
x._UnPack(unpackOptions)
|
|
return x
|
|
|
|
# UnpackOptionsT
|
|
def _UnPack(self, unpackOptions):
|
|
if unpackOptions is None:
|
|
return
|
|
self.num = unpackOptions.Num()
|
|
self.axis = unpackOptions.Axis()
|
|
|
|
# UnpackOptionsT
|
|
def Pack(self, builder):
|
|
UnpackOptionsStart(builder)
|
|
UnpackOptionsAddNum(builder, self.num)
|
|
UnpackOptionsAddAxis(builder, self.axis)
|
|
unpackOptions = UnpackOptionsEnd(builder)
|
|
return unpackOptions
|
|
|
|
|
|
class FloorDivOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = FloorDivOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsFloorDivOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def FloorDivOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# FloorDivOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def FloorDivOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def FloorDivOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class FloorDivOptionsT(object):
|
|
|
|
# FloorDivOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
floorDivOptions = FloorDivOptions()
|
|
floorDivOptions.Init(buf, pos)
|
|
return cls.InitFromObj(floorDivOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, floorDivOptions):
|
|
x = FloorDivOptionsT()
|
|
x._UnPack(floorDivOptions)
|
|
return x
|
|
|
|
# FloorDivOptionsT
|
|
def _UnPack(self, floorDivOptions):
|
|
if floorDivOptions is None:
|
|
return
|
|
|
|
# FloorDivOptionsT
|
|
def Pack(self, builder):
|
|
FloorDivOptionsStart(builder)
|
|
floorDivOptions = FloorDivOptionsEnd(builder)
|
|
return floorDivOptions
|
|
|
|
|
|
class SquareOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SquareOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSquareOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SquareOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SquareOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SquareOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SquareOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SquareOptionsT(object):
|
|
|
|
# SquareOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
squareOptions = SquareOptions()
|
|
squareOptions.Init(buf, pos)
|
|
return cls.InitFromObj(squareOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, squareOptions):
|
|
x = SquareOptionsT()
|
|
x._UnPack(squareOptions)
|
|
return x
|
|
|
|
# SquareOptionsT
|
|
def _UnPack(self, squareOptions):
|
|
if squareOptions is None:
|
|
return
|
|
|
|
# SquareOptionsT
|
|
def Pack(self, builder):
|
|
SquareOptionsStart(builder)
|
|
squareOptions = SquareOptionsEnd(builder)
|
|
return squareOptions
|
|
|
|
|
|
class ZerosLikeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ZerosLikeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsZerosLikeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ZerosLikeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ZerosLikeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ZerosLikeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ZerosLikeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ZerosLikeOptionsT(object):
|
|
|
|
# ZerosLikeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
zerosLikeOptions = ZerosLikeOptions()
|
|
zerosLikeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(zerosLikeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, zerosLikeOptions):
|
|
x = ZerosLikeOptionsT()
|
|
x._UnPack(zerosLikeOptions)
|
|
return x
|
|
|
|
# ZerosLikeOptionsT
|
|
def _UnPack(self, zerosLikeOptions):
|
|
if zerosLikeOptions is None:
|
|
return
|
|
|
|
# ZerosLikeOptionsT
|
|
def Pack(self, builder):
|
|
ZerosLikeOptionsStart(builder)
|
|
zerosLikeOptions = ZerosLikeOptionsEnd(builder)
|
|
return zerosLikeOptions
|
|
|
|
|
|
class FillOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = FillOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsFillOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def FillOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# FillOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def FillOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def FillOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class FillOptionsT(object):
|
|
|
|
# FillOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
fillOptions = FillOptions()
|
|
fillOptions.Init(buf, pos)
|
|
return cls.InitFromObj(fillOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, fillOptions):
|
|
x = FillOptionsT()
|
|
x._UnPack(fillOptions)
|
|
return x
|
|
|
|
# FillOptionsT
|
|
def _UnPack(self, fillOptions):
|
|
if fillOptions is None:
|
|
return
|
|
|
|
# FillOptionsT
|
|
def Pack(self, builder):
|
|
FillOptionsStart(builder)
|
|
fillOptions = FillOptionsEnd(builder)
|
|
return fillOptions
|
|
|
|
|
|
class FloorModOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = FloorModOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsFloorModOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def FloorModOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# FloorModOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def FloorModOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def FloorModOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class FloorModOptionsT(object):
|
|
|
|
# FloorModOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
floorModOptions = FloorModOptions()
|
|
floorModOptions.Init(buf, pos)
|
|
return cls.InitFromObj(floorModOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, floorModOptions):
|
|
x = FloorModOptionsT()
|
|
x._UnPack(floorModOptions)
|
|
return x
|
|
|
|
# FloorModOptionsT
|
|
def _UnPack(self, floorModOptions):
|
|
if floorModOptions is None:
|
|
return
|
|
|
|
# FloorModOptionsT
|
|
def Pack(self, builder):
|
|
FloorModOptionsStart(builder)
|
|
floorModOptions = FloorModOptionsEnd(builder)
|
|
return floorModOptions
|
|
|
|
|
|
class RangeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = RangeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRangeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def RangeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# RangeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def RangeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def RangeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class RangeOptionsT(object):
|
|
|
|
# RangeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
rangeOptions = RangeOptions()
|
|
rangeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(rangeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, rangeOptions):
|
|
x = RangeOptionsT()
|
|
x._UnPack(rangeOptions)
|
|
return x
|
|
|
|
# RangeOptionsT
|
|
def _UnPack(self, rangeOptions):
|
|
if rangeOptions is None:
|
|
return
|
|
|
|
# RangeOptionsT
|
|
def Pack(self, builder):
|
|
RangeOptionsStart(builder)
|
|
rangeOptions = RangeOptionsEnd(builder)
|
|
return rangeOptions
|
|
|
|
|
|
class LeakyReluOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = LeakyReluOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsLeakyReluOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def LeakyReluOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# LeakyReluOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# LeakyReluOptions
|
|
def Alpha(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
return 0.0
|
|
|
|
def LeakyReluOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def LeakyReluOptionsAddAlpha(builder, alpha):
|
|
builder.PrependFloat32Slot(0, alpha, 0.0)
|
|
|
|
def LeakyReluOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class LeakyReluOptionsT(object):
|
|
|
|
# LeakyReluOptionsT
|
|
def __init__(self):
|
|
self.alpha = 0.0 # type: float
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
leakyReluOptions = LeakyReluOptions()
|
|
leakyReluOptions.Init(buf, pos)
|
|
return cls.InitFromObj(leakyReluOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, leakyReluOptions):
|
|
x = LeakyReluOptionsT()
|
|
x._UnPack(leakyReluOptions)
|
|
return x
|
|
|
|
# LeakyReluOptionsT
|
|
def _UnPack(self, leakyReluOptions):
|
|
if leakyReluOptions is None:
|
|
return
|
|
self.alpha = leakyReluOptions.Alpha()
|
|
|
|
# LeakyReluOptionsT
|
|
def Pack(self, builder):
|
|
LeakyReluOptionsStart(builder)
|
|
LeakyReluOptionsAddAlpha(builder, self.alpha)
|
|
leakyReluOptions = LeakyReluOptionsEnd(builder)
|
|
return leakyReluOptions
|
|
|
|
|
|
class SquaredDifferenceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SquaredDifferenceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSquaredDifferenceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SquaredDifferenceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SquaredDifferenceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SquaredDifferenceOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SquaredDifferenceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SquaredDifferenceOptionsT(object):
|
|
|
|
# SquaredDifferenceOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
squaredDifferenceOptions = SquaredDifferenceOptions()
|
|
squaredDifferenceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(squaredDifferenceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, squaredDifferenceOptions):
|
|
x = SquaredDifferenceOptionsT()
|
|
x._UnPack(squaredDifferenceOptions)
|
|
return x
|
|
|
|
# SquaredDifferenceOptionsT
|
|
def _UnPack(self, squaredDifferenceOptions):
|
|
if squaredDifferenceOptions is None:
|
|
return
|
|
|
|
# SquaredDifferenceOptionsT
|
|
def Pack(self, builder):
|
|
SquaredDifferenceOptionsStart(builder)
|
|
squaredDifferenceOptions = SquaredDifferenceOptionsEnd(builder)
|
|
return squaredDifferenceOptions
|
|
|
|
|
|
class MirrorPadOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = MirrorPadOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMirrorPadOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MirrorPadOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# MirrorPadOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# MirrorPadOptions
|
|
def Mode(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def MirrorPadOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def MirrorPadOptionsAddMode(builder, mode):
|
|
builder.PrependInt8Slot(0, mode, 0)
|
|
|
|
def MirrorPadOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MirrorPadOptionsT(object):
|
|
|
|
# MirrorPadOptionsT
|
|
def __init__(self):
|
|
self.mode = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
mirrorPadOptions = MirrorPadOptions()
|
|
mirrorPadOptions.Init(buf, pos)
|
|
return cls.InitFromObj(mirrorPadOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, mirrorPadOptions):
|
|
x = MirrorPadOptionsT()
|
|
x._UnPack(mirrorPadOptions)
|
|
return x
|
|
|
|
# MirrorPadOptionsT
|
|
def _UnPack(self, mirrorPadOptions):
|
|
if mirrorPadOptions is None:
|
|
return
|
|
self.mode = mirrorPadOptions.Mode()
|
|
|
|
# MirrorPadOptionsT
|
|
def Pack(self, builder):
|
|
MirrorPadOptionsStart(builder)
|
|
MirrorPadOptionsAddMode(builder, self.mode)
|
|
mirrorPadOptions = MirrorPadOptionsEnd(builder)
|
|
return mirrorPadOptions
|
|
|
|
|
|
class UniqueOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UniqueOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUniqueOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UniqueOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UniqueOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# UniqueOptions
|
|
def IdxOutType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 2
|
|
|
|
def UniqueOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def UniqueOptionsAddIdxOutType(builder, idxOutType):
|
|
builder.PrependInt8Slot(0, idxOutType, 2)
|
|
|
|
def UniqueOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UniqueOptionsT(object):
|
|
|
|
# UniqueOptionsT
|
|
def __init__(self):
|
|
self.idxOutType = 2 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
uniqueOptions = UniqueOptions()
|
|
uniqueOptions.Init(buf, pos)
|
|
return cls.InitFromObj(uniqueOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, uniqueOptions):
|
|
x = UniqueOptionsT()
|
|
x._UnPack(uniqueOptions)
|
|
return x
|
|
|
|
# UniqueOptionsT
|
|
def _UnPack(self, uniqueOptions):
|
|
if uniqueOptions is None:
|
|
return
|
|
self.idxOutType = uniqueOptions.IdxOutType()
|
|
|
|
# UniqueOptionsT
|
|
def Pack(self, builder):
|
|
UniqueOptionsStart(builder)
|
|
UniqueOptionsAddIdxOutType(builder, self.idxOutType)
|
|
uniqueOptions = UniqueOptionsEnd(builder)
|
|
return uniqueOptions
|
|
|
|
|
|
class ReverseV2Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReverseV2Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReverseV2Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReverseV2OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReverseV2Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ReverseV2OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ReverseV2OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ReverseV2OptionsT(object):
|
|
|
|
# ReverseV2OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
reverseV2Options = ReverseV2Options()
|
|
reverseV2Options.Init(buf, pos)
|
|
return cls.InitFromObj(reverseV2Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, reverseV2Options):
|
|
x = ReverseV2OptionsT()
|
|
x._UnPack(reverseV2Options)
|
|
return x
|
|
|
|
# ReverseV2OptionsT
|
|
def _UnPack(self, reverseV2Options):
|
|
if reverseV2Options is None:
|
|
return
|
|
|
|
# ReverseV2OptionsT
|
|
def Pack(self, builder):
|
|
ReverseV2OptionsStart(builder)
|
|
reverseV2Options = ReverseV2OptionsEnd(builder)
|
|
return reverseV2Options
|
|
|
|
|
|
class AddNOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = AddNOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsAddNOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def AddNOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# AddNOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def AddNOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def AddNOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class AddNOptionsT(object):
|
|
|
|
# AddNOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
addNoptions = AddNOptions()
|
|
addNoptions.Init(buf, pos)
|
|
return cls.InitFromObj(addNoptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, addNoptions):
|
|
x = AddNOptionsT()
|
|
x._UnPack(addNoptions)
|
|
return x
|
|
|
|
# AddNOptionsT
|
|
def _UnPack(self, addNoptions):
|
|
if addNoptions is None:
|
|
return
|
|
|
|
# AddNOptionsT
|
|
def Pack(self, builder):
|
|
AddNOptionsStart(builder)
|
|
addNoptions = AddNOptionsEnd(builder)
|
|
return addNoptions
|
|
|
|
|
|
class GatherNdOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = GatherNdOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsGatherNdOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def GatherNdOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# GatherNdOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def GatherNdOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def GatherNdOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class GatherNdOptionsT(object):
|
|
|
|
# GatherNdOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
gatherNdOptions = GatherNdOptions()
|
|
gatherNdOptions.Init(buf, pos)
|
|
return cls.InitFromObj(gatherNdOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, gatherNdOptions):
|
|
x = GatherNdOptionsT()
|
|
x._UnPack(gatherNdOptions)
|
|
return x
|
|
|
|
# GatherNdOptionsT
|
|
def _UnPack(self, gatherNdOptions):
|
|
if gatherNdOptions is None:
|
|
return
|
|
|
|
# GatherNdOptionsT
|
|
def Pack(self, builder):
|
|
GatherNdOptionsStart(builder)
|
|
gatherNdOptions = GatherNdOptionsEnd(builder)
|
|
return gatherNdOptions
|
|
|
|
|
|
class WhereOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = WhereOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsWhereOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def WhereOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# WhereOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def WhereOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def WhereOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class WhereOptionsT(object):
|
|
|
|
# WhereOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
whereOptions = WhereOptions()
|
|
whereOptions.Init(buf, pos)
|
|
return cls.InitFromObj(whereOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, whereOptions):
|
|
x = WhereOptionsT()
|
|
x._UnPack(whereOptions)
|
|
return x
|
|
|
|
# WhereOptionsT
|
|
def _UnPack(self, whereOptions):
|
|
if whereOptions is None:
|
|
return
|
|
|
|
# WhereOptionsT
|
|
def Pack(self, builder):
|
|
WhereOptionsStart(builder)
|
|
whereOptions = WhereOptionsEnd(builder)
|
|
return whereOptions
|
|
|
|
|
|
class ReverseSequenceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReverseSequenceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReverseSequenceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReverseSequenceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReverseSequenceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ReverseSequenceOptions
|
|
def SeqDim(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# ReverseSequenceOptions
|
|
def BatchDim(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ReverseSequenceOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def ReverseSequenceOptionsAddSeqDim(builder, seqDim):
|
|
builder.PrependInt32Slot(0, seqDim, 0)
|
|
|
|
def ReverseSequenceOptionsAddBatchDim(builder, batchDim):
|
|
builder.PrependInt32Slot(1, batchDim, 0)
|
|
|
|
def ReverseSequenceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ReverseSequenceOptionsT(object):
|
|
|
|
# ReverseSequenceOptionsT
|
|
def __init__(self):
|
|
self.seqDim = 0 # type: int
|
|
self.batchDim = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
reverseSequenceOptions = ReverseSequenceOptions()
|
|
reverseSequenceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(reverseSequenceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, reverseSequenceOptions):
|
|
x = ReverseSequenceOptionsT()
|
|
x._UnPack(reverseSequenceOptions)
|
|
return x
|
|
|
|
# ReverseSequenceOptionsT
|
|
def _UnPack(self, reverseSequenceOptions):
|
|
if reverseSequenceOptions is None:
|
|
return
|
|
self.seqDim = reverseSequenceOptions.SeqDim()
|
|
self.batchDim = reverseSequenceOptions.BatchDim()
|
|
|
|
# ReverseSequenceOptionsT
|
|
def Pack(self, builder):
|
|
ReverseSequenceOptionsStart(builder)
|
|
ReverseSequenceOptionsAddSeqDim(builder, self.seqDim)
|
|
ReverseSequenceOptionsAddBatchDim(builder, self.batchDim)
|
|
reverseSequenceOptions = ReverseSequenceOptionsEnd(builder)
|
|
return reverseSequenceOptions
|
|
|
|
|
|
class MatrixDiagOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = MatrixDiagOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMatrixDiagOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MatrixDiagOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# MatrixDiagOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def MatrixDiagOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def MatrixDiagOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MatrixDiagOptionsT(object):
|
|
|
|
# MatrixDiagOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
matrixDiagOptions = MatrixDiagOptions()
|
|
matrixDiagOptions.Init(buf, pos)
|
|
return cls.InitFromObj(matrixDiagOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, matrixDiagOptions):
|
|
x = MatrixDiagOptionsT()
|
|
x._UnPack(matrixDiagOptions)
|
|
return x
|
|
|
|
# MatrixDiagOptionsT
|
|
def _UnPack(self, matrixDiagOptions):
|
|
if matrixDiagOptions is None:
|
|
return
|
|
|
|
# MatrixDiagOptionsT
|
|
def Pack(self, builder):
|
|
MatrixDiagOptionsStart(builder)
|
|
matrixDiagOptions = MatrixDiagOptionsEnd(builder)
|
|
return matrixDiagOptions
|
|
|
|
|
|
class QuantizeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = QuantizeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsQuantizeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def QuantizeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# QuantizeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def QuantizeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def QuantizeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class QuantizeOptionsT(object):
|
|
|
|
# QuantizeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
quantizeOptions = QuantizeOptions()
|
|
quantizeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(quantizeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, quantizeOptions):
|
|
x = QuantizeOptionsT()
|
|
x._UnPack(quantizeOptions)
|
|
return x
|
|
|
|
# QuantizeOptionsT
|
|
def _UnPack(self, quantizeOptions):
|
|
if quantizeOptions is None:
|
|
return
|
|
|
|
# QuantizeOptionsT
|
|
def Pack(self, builder):
|
|
QuantizeOptionsStart(builder)
|
|
quantizeOptions = QuantizeOptionsEnd(builder)
|
|
return quantizeOptions
|
|
|
|
|
|
class MatrixSetDiagOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = MatrixSetDiagOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMatrixSetDiagOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MatrixSetDiagOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# MatrixSetDiagOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def MatrixSetDiagOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def MatrixSetDiagOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MatrixSetDiagOptionsT(object):
|
|
|
|
# MatrixSetDiagOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
matrixSetDiagOptions = MatrixSetDiagOptions()
|
|
matrixSetDiagOptions.Init(buf, pos)
|
|
return cls.InitFromObj(matrixSetDiagOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, matrixSetDiagOptions):
|
|
x = MatrixSetDiagOptionsT()
|
|
x._UnPack(matrixSetDiagOptions)
|
|
return x
|
|
|
|
# MatrixSetDiagOptionsT
|
|
def _UnPack(self, matrixSetDiagOptions):
|
|
if matrixSetDiagOptions is None:
|
|
return
|
|
|
|
# MatrixSetDiagOptionsT
|
|
def Pack(self, builder):
|
|
MatrixSetDiagOptionsStart(builder)
|
|
matrixSetDiagOptions = MatrixSetDiagOptionsEnd(builder)
|
|
return matrixSetDiagOptions
|
|
|
|
|
|
class IfOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = IfOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsIfOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def IfOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# IfOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# IfOptions
|
|
def ThenSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# IfOptions
|
|
def ElseSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def IfOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def IfOptionsAddThenSubgraphIndex(builder, thenSubgraphIndex):
|
|
builder.PrependInt32Slot(0, thenSubgraphIndex, 0)
|
|
|
|
def IfOptionsAddElseSubgraphIndex(builder, elseSubgraphIndex):
|
|
builder.PrependInt32Slot(1, elseSubgraphIndex, 0)
|
|
|
|
def IfOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class IfOptionsT(object):
|
|
|
|
# IfOptionsT
|
|
def __init__(self):
|
|
self.thenSubgraphIndex = 0 # type: int
|
|
self.elseSubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
ifOptions = IfOptions()
|
|
ifOptions.Init(buf, pos)
|
|
return cls.InitFromObj(ifOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, ifOptions):
|
|
x = IfOptionsT()
|
|
x._UnPack(ifOptions)
|
|
return x
|
|
|
|
# IfOptionsT
|
|
def _UnPack(self, ifOptions):
|
|
if ifOptions is None:
|
|
return
|
|
self.thenSubgraphIndex = ifOptions.ThenSubgraphIndex()
|
|
self.elseSubgraphIndex = ifOptions.ElseSubgraphIndex()
|
|
|
|
# IfOptionsT
|
|
def Pack(self, builder):
|
|
IfOptionsStart(builder)
|
|
IfOptionsAddThenSubgraphIndex(builder, self.thenSubgraphIndex)
|
|
IfOptionsAddElseSubgraphIndex(builder, self.elseSubgraphIndex)
|
|
ifOptions = IfOptionsEnd(builder)
|
|
return ifOptions
|
|
|
|
|
|
class CallOnceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CallOnceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCallOnceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CallOnceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CallOnceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# CallOnceOptions
|
|
def InitSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def CallOnceOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def CallOnceOptionsAddInitSubgraphIndex(builder, initSubgraphIndex):
|
|
builder.PrependInt32Slot(0, initSubgraphIndex, 0)
|
|
|
|
def CallOnceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class CallOnceOptionsT(object):
|
|
|
|
# CallOnceOptionsT
|
|
def __init__(self):
|
|
self.initSubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
callOnceOptions = CallOnceOptions()
|
|
callOnceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(callOnceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, callOnceOptions):
|
|
x = CallOnceOptionsT()
|
|
x._UnPack(callOnceOptions)
|
|
return x
|
|
|
|
# CallOnceOptionsT
|
|
def _UnPack(self, callOnceOptions):
|
|
if callOnceOptions is None:
|
|
return
|
|
self.initSubgraphIndex = callOnceOptions.InitSubgraphIndex()
|
|
|
|
# CallOnceOptionsT
|
|
def Pack(self, builder):
|
|
CallOnceOptionsStart(builder)
|
|
CallOnceOptionsAddInitSubgraphIndex(builder, self.initSubgraphIndex)
|
|
callOnceOptions = CallOnceOptionsEnd(builder)
|
|
return callOnceOptions
|
|
|
|
|
|
class WhileOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = WhileOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsWhileOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def WhileOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# WhileOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# WhileOptions
|
|
def CondSubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# WhileOptions
|
|
def BodySubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def WhileOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def WhileOptionsAddCondSubgraphIndex(builder, condSubgraphIndex):
|
|
builder.PrependInt32Slot(0, condSubgraphIndex, 0)
|
|
|
|
def WhileOptionsAddBodySubgraphIndex(builder, bodySubgraphIndex):
|
|
builder.PrependInt32Slot(1, bodySubgraphIndex, 0)
|
|
|
|
def WhileOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class WhileOptionsT(object):
|
|
|
|
# WhileOptionsT
|
|
def __init__(self):
|
|
self.condSubgraphIndex = 0 # type: int
|
|
self.bodySubgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
whileOptions = WhileOptions()
|
|
whileOptions.Init(buf, pos)
|
|
return cls.InitFromObj(whileOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, whileOptions):
|
|
x = WhileOptionsT()
|
|
x._UnPack(whileOptions)
|
|
return x
|
|
|
|
# WhileOptionsT
|
|
def _UnPack(self, whileOptions):
|
|
if whileOptions is None:
|
|
return
|
|
self.condSubgraphIndex = whileOptions.CondSubgraphIndex()
|
|
self.bodySubgraphIndex = whileOptions.BodySubgraphIndex()
|
|
|
|
# WhileOptionsT
|
|
def Pack(self, builder):
|
|
WhileOptionsStart(builder)
|
|
WhileOptionsAddCondSubgraphIndex(builder, self.condSubgraphIndex)
|
|
WhileOptionsAddBodySubgraphIndex(builder, self.bodySubgraphIndex)
|
|
whileOptions = WhileOptionsEnd(builder)
|
|
return whileOptions
|
|
|
|
|
|
class NonMaxSuppressionV4Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = NonMaxSuppressionV4Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsNonMaxSuppressionV4Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def NonMaxSuppressionV4OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# NonMaxSuppressionV4Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def NonMaxSuppressionV4OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def NonMaxSuppressionV4OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class NonMaxSuppressionV4OptionsT(object):
|
|
|
|
# NonMaxSuppressionV4OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
nonMaxSuppressionV4Options = NonMaxSuppressionV4Options()
|
|
nonMaxSuppressionV4Options.Init(buf, pos)
|
|
return cls.InitFromObj(nonMaxSuppressionV4Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, nonMaxSuppressionV4Options):
|
|
x = NonMaxSuppressionV4OptionsT()
|
|
x._UnPack(nonMaxSuppressionV4Options)
|
|
return x
|
|
|
|
# NonMaxSuppressionV4OptionsT
|
|
def _UnPack(self, nonMaxSuppressionV4Options):
|
|
if nonMaxSuppressionV4Options is None:
|
|
return
|
|
|
|
# NonMaxSuppressionV4OptionsT
|
|
def Pack(self, builder):
|
|
NonMaxSuppressionV4OptionsStart(builder)
|
|
nonMaxSuppressionV4Options = NonMaxSuppressionV4OptionsEnd(builder)
|
|
return nonMaxSuppressionV4Options
|
|
|
|
|
|
class NonMaxSuppressionV5Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = NonMaxSuppressionV5Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsNonMaxSuppressionV5Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def NonMaxSuppressionV5OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# NonMaxSuppressionV5Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def NonMaxSuppressionV5OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def NonMaxSuppressionV5OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class NonMaxSuppressionV5OptionsT(object):
|
|
|
|
# NonMaxSuppressionV5OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
nonMaxSuppressionV5Options = NonMaxSuppressionV5Options()
|
|
nonMaxSuppressionV5Options.Init(buf, pos)
|
|
return cls.InitFromObj(nonMaxSuppressionV5Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, nonMaxSuppressionV5Options):
|
|
x = NonMaxSuppressionV5OptionsT()
|
|
x._UnPack(nonMaxSuppressionV5Options)
|
|
return x
|
|
|
|
# NonMaxSuppressionV5OptionsT
|
|
def _UnPack(self, nonMaxSuppressionV5Options):
|
|
if nonMaxSuppressionV5Options is None:
|
|
return
|
|
|
|
# NonMaxSuppressionV5OptionsT
|
|
def Pack(self, builder):
|
|
NonMaxSuppressionV5OptionsStart(builder)
|
|
nonMaxSuppressionV5Options = NonMaxSuppressionV5OptionsEnd(builder)
|
|
return nonMaxSuppressionV5Options
|
|
|
|
|
|
class ScatterNdOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ScatterNdOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsScatterNdOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ScatterNdOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ScatterNdOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ScatterNdOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ScatterNdOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ScatterNdOptionsT(object):
|
|
|
|
# ScatterNdOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
scatterNdOptions = ScatterNdOptions()
|
|
scatterNdOptions.Init(buf, pos)
|
|
return cls.InitFromObj(scatterNdOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, scatterNdOptions):
|
|
x = ScatterNdOptionsT()
|
|
x._UnPack(scatterNdOptions)
|
|
return x
|
|
|
|
# ScatterNdOptionsT
|
|
def _UnPack(self, scatterNdOptions):
|
|
if scatterNdOptions is None:
|
|
return
|
|
|
|
# ScatterNdOptionsT
|
|
def Pack(self, builder):
|
|
ScatterNdOptionsStart(builder)
|
|
scatterNdOptions = ScatterNdOptionsEnd(builder)
|
|
return scatterNdOptions
|
|
|
|
|
|
class SelectV2Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SelectV2Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSelectV2Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SelectV2OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SelectV2Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SelectV2OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SelectV2OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SelectV2OptionsT(object):
|
|
|
|
# SelectV2OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
selectV2Options = SelectV2Options()
|
|
selectV2Options.Init(buf, pos)
|
|
return cls.InitFromObj(selectV2Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, selectV2Options):
|
|
x = SelectV2OptionsT()
|
|
x._UnPack(selectV2Options)
|
|
return x
|
|
|
|
# SelectV2OptionsT
|
|
def _UnPack(self, selectV2Options):
|
|
if selectV2Options is None:
|
|
return
|
|
|
|
# SelectV2OptionsT
|
|
def Pack(self, builder):
|
|
SelectV2OptionsStart(builder)
|
|
selectV2Options = SelectV2OptionsEnd(builder)
|
|
return selectV2Options
|
|
|
|
|
|
class DensifyOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DensifyOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDensifyOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DensifyOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DensifyOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def DensifyOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def DensifyOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DensifyOptionsT(object):
|
|
|
|
# DensifyOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
densifyOptions = DensifyOptions()
|
|
densifyOptions.Init(buf, pos)
|
|
return cls.InitFromObj(densifyOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, densifyOptions):
|
|
x = DensifyOptionsT()
|
|
x._UnPack(densifyOptions)
|
|
return x
|
|
|
|
# DensifyOptionsT
|
|
def _UnPack(self, densifyOptions):
|
|
if densifyOptions is None:
|
|
return
|
|
|
|
# DensifyOptionsT
|
|
def Pack(self, builder):
|
|
DensifyOptionsStart(builder)
|
|
densifyOptions = DensifyOptionsEnd(builder)
|
|
return densifyOptions
|
|
|
|
|
|
class SegmentSumOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SegmentSumOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSegmentSumOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SegmentSumOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SegmentSumOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SegmentSumOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SegmentSumOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SegmentSumOptionsT(object):
|
|
|
|
# SegmentSumOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
segmentSumOptions = SegmentSumOptions()
|
|
segmentSumOptions.Init(buf, pos)
|
|
return cls.InitFromObj(segmentSumOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, segmentSumOptions):
|
|
x = SegmentSumOptionsT()
|
|
x._UnPack(segmentSumOptions)
|
|
return x
|
|
|
|
# SegmentSumOptionsT
|
|
def _UnPack(self, segmentSumOptions):
|
|
if segmentSumOptions is None:
|
|
return
|
|
|
|
# SegmentSumOptionsT
|
|
def Pack(self, builder):
|
|
SegmentSumOptionsStart(builder)
|
|
segmentSumOptions = SegmentSumOptionsEnd(builder)
|
|
return segmentSumOptions
|
|
|
|
|
|
class BatchMatMulOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BatchMatMulOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBatchMatMulOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BatchMatMulOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BatchMatMulOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# BatchMatMulOptions
|
|
def AdjX(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# BatchMatMulOptions
|
|
def AdjY(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# BatchMatMulOptions
|
|
def AsymmetricQuantizeInputs(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def BatchMatMulOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def BatchMatMulOptionsAddAdjX(builder, adjX):
|
|
builder.PrependBoolSlot(0, adjX, 0)
|
|
|
|
def BatchMatMulOptionsAddAdjY(builder, adjY):
|
|
builder.PrependBoolSlot(1, adjY, 0)
|
|
|
|
def BatchMatMulOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
|
|
builder.PrependBoolSlot(2, asymmetricQuantizeInputs, 0)
|
|
|
|
def BatchMatMulOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BatchMatMulOptionsT(object):
|
|
|
|
# BatchMatMulOptionsT
|
|
def __init__(self):
|
|
self.adjX = False # type: bool
|
|
self.adjY = False # type: bool
|
|
self.asymmetricQuantizeInputs = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
batchMatMulOptions = BatchMatMulOptions()
|
|
batchMatMulOptions.Init(buf, pos)
|
|
return cls.InitFromObj(batchMatMulOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, batchMatMulOptions):
|
|
x = BatchMatMulOptionsT()
|
|
x._UnPack(batchMatMulOptions)
|
|
return x
|
|
|
|
# BatchMatMulOptionsT
|
|
def _UnPack(self, batchMatMulOptions):
|
|
if batchMatMulOptions is None:
|
|
return
|
|
self.adjX = batchMatMulOptions.AdjX()
|
|
self.adjY = batchMatMulOptions.AdjY()
|
|
self.asymmetricQuantizeInputs = batchMatMulOptions.AsymmetricQuantizeInputs()
|
|
|
|
# BatchMatMulOptionsT
|
|
def Pack(self, builder):
|
|
BatchMatMulOptionsStart(builder)
|
|
BatchMatMulOptionsAddAdjX(builder, self.adjX)
|
|
BatchMatMulOptionsAddAdjY(builder, self.adjY)
|
|
BatchMatMulOptionsAddAsymmetricQuantizeInputs(builder, self.asymmetricQuantizeInputs)
|
|
batchMatMulOptions = BatchMatMulOptionsEnd(builder)
|
|
return batchMatMulOptions
|
|
|
|
|
|
class CumsumOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = CumsumOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsCumsumOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def CumsumOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# CumsumOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# CumsumOptions
|
|
def Exclusive(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
# CumsumOptions
|
|
def Reverse(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def CumsumOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def CumsumOptionsAddExclusive(builder, exclusive):
|
|
builder.PrependBoolSlot(0, exclusive, 0)
|
|
|
|
def CumsumOptionsAddReverse(builder, reverse):
|
|
builder.PrependBoolSlot(1, reverse, 0)
|
|
|
|
def CumsumOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class CumsumOptionsT(object):
|
|
|
|
# CumsumOptionsT
|
|
def __init__(self):
|
|
self.exclusive = False # type: bool
|
|
self.reverse = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
cumsumOptions = CumsumOptions()
|
|
cumsumOptions.Init(buf, pos)
|
|
return cls.InitFromObj(cumsumOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, cumsumOptions):
|
|
x = CumsumOptionsT()
|
|
x._UnPack(cumsumOptions)
|
|
return x
|
|
|
|
# CumsumOptionsT
|
|
def _UnPack(self, cumsumOptions):
|
|
if cumsumOptions is None:
|
|
return
|
|
self.exclusive = cumsumOptions.Exclusive()
|
|
self.reverse = cumsumOptions.Reverse()
|
|
|
|
# CumsumOptionsT
|
|
def Pack(self, builder):
|
|
CumsumOptionsStart(builder)
|
|
CumsumOptionsAddExclusive(builder, self.exclusive)
|
|
CumsumOptionsAddReverse(builder, self.reverse)
|
|
cumsumOptions = CumsumOptionsEnd(builder)
|
|
return cumsumOptions
|
|
|
|
|
|
class BroadcastToOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BroadcastToOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBroadcastToOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BroadcastToOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BroadcastToOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def BroadcastToOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def BroadcastToOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BroadcastToOptionsT(object):
|
|
|
|
# BroadcastToOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
broadcastToOptions = BroadcastToOptions()
|
|
broadcastToOptions.Init(buf, pos)
|
|
return cls.InitFromObj(broadcastToOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, broadcastToOptions):
|
|
x = BroadcastToOptionsT()
|
|
x._UnPack(broadcastToOptions)
|
|
return x
|
|
|
|
# BroadcastToOptionsT
|
|
def _UnPack(self, broadcastToOptions):
|
|
if broadcastToOptions is None:
|
|
return
|
|
|
|
# BroadcastToOptionsT
|
|
def Pack(self, builder):
|
|
BroadcastToOptionsStart(builder)
|
|
broadcastToOptions = BroadcastToOptionsEnd(builder)
|
|
return broadcastToOptions
|
|
|
|
|
|
class Rfft2dOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Rfft2dOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRfft2dOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def Rfft2dOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Rfft2dOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def Rfft2dOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def Rfft2dOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class Rfft2dOptionsT(object):
|
|
|
|
# Rfft2dOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
rfft2dOptions = Rfft2dOptions()
|
|
rfft2dOptions.Init(buf, pos)
|
|
return cls.InitFromObj(rfft2dOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, rfft2dOptions):
|
|
x = Rfft2dOptionsT()
|
|
x._UnPack(rfft2dOptions)
|
|
return x
|
|
|
|
# Rfft2dOptionsT
|
|
def _UnPack(self, rfft2dOptions):
|
|
if rfft2dOptions is None:
|
|
return
|
|
|
|
# Rfft2dOptionsT
|
|
def Pack(self, builder):
|
|
Rfft2dOptionsStart(builder)
|
|
rfft2dOptions = Rfft2dOptionsEnd(builder)
|
|
return rfft2dOptions
|
|
|
|
|
|
class HashtableOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = HashtableOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsHashtableOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def HashtableOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# HashtableOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# HashtableOptions
|
|
def TableId(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# HashtableOptions
|
|
def KeyDtype(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# HashtableOptions
|
|
def ValueDtype(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def HashtableOptionsStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def HashtableOptionsAddTableId(builder, tableId):
|
|
builder.PrependInt32Slot(0, tableId, 0)
|
|
|
|
def HashtableOptionsAddKeyDtype(builder, keyDtype):
|
|
builder.PrependInt8Slot(1, keyDtype, 0)
|
|
|
|
def HashtableOptionsAddValueDtype(builder, valueDtype):
|
|
builder.PrependInt8Slot(2, valueDtype, 0)
|
|
|
|
def HashtableOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class HashtableOptionsT(object):
|
|
|
|
# HashtableOptionsT
|
|
def __init__(self):
|
|
self.tableId = 0 # type: int
|
|
self.keyDtype = 0 # type: int
|
|
self.valueDtype = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
hashtableOptions = HashtableOptions()
|
|
hashtableOptions.Init(buf, pos)
|
|
return cls.InitFromObj(hashtableOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, hashtableOptions):
|
|
x = HashtableOptionsT()
|
|
x._UnPack(hashtableOptions)
|
|
return x
|
|
|
|
# HashtableOptionsT
|
|
def _UnPack(self, hashtableOptions):
|
|
if hashtableOptions is None:
|
|
return
|
|
self.tableId = hashtableOptions.TableId()
|
|
self.keyDtype = hashtableOptions.KeyDtype()
|
|
self.valueDtype = hashtableOptions.ValueDtype()
|
|
|
|
# HashtableOptionsT
|
|
def Pack(self, builder):
|
|
HashtableOptionsStart(builder)
|
|
HashtableOptionsAddTableId(builder, self.tableId)
|
|
HashtableOptionsAddKeyDtype(builder, self.keyDtype)
|
|
HashtableOptionsAddValueDtype(builder, self.valueDtype)
|
|
hashtableOptions = HashtableOptionsEnd(builder)
|
|
return hashtableOptions
|
|
|
|
|
|
class HashtableFindOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = HashtableFindOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsHashtableFindOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def HashtableFindOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# HashtableFindOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def HashtableFindOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def HashtableFindOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class HashtableFindOptionsT(object):
|
|
|
|
# HashtableFindOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
hashtableFindOptions = HashtableFindOptions()
|
|
hashtableFindOptions.Init(buf, pos)
|
|
return cls.InitFromObj(hashtableFindOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, hashtableFindOptions):
|
|
x = HashtableFindOptionsT()
|
|
x._UnPack(hashtableFindOptions)
|
|
return x
|
|
|
|
# HashtableFindOptionsT
|
|
def _UnPack(self, hashtableFindOptions):
|
|
if hashtableFindOptions is None:
|
|
return
|
|
|
|
# HashtableFindOptionsT
|
|
def Pack(self, builder):
|
|
HashtableFindOptionsStart(builder)
|
|
hashtableFindOptions = HashtableFindOptionsEnd(builder)
|
|
return hashtableFindOptions
|
|
|
|
|
|
class HashtableImportOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = HashtableImportOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsHashtableImportOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def HashtableImportOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# HashtableImportOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def HashtableImportOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def HashtableImportOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class HashtableImportOptionsT(object):
|
|
|
|
# HashtableImportOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
hashtableImportOptions = HashtableImportOptions()
|
|
hashtableImportOptions.Init(buf, pos)
|
|
return cls.InitFromObj(hashtableImportOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, hashtableImportOptions):
|
|
x = HashtableImportOptionsT()
|
|
x._UnPack(hashtableImportOptions)
|
|
return x
|
|
|
|
# HashtableImportOptionsT
|
|
def _UnPack(self, hashtableImportOptions):
|
|
if hashtableImportOptions is None:
|
|
return
|
|
|
|
# HashtableImportOptionsT
|
|
def Pack(self, builder):
|
|
HashtableImportOptionsStart(builder)
|
|
hashtableImportOptions = HashtableImportOptionsEnd(builder)
|
|
return hashtableImportOptions
|
|
|
|
|
|
class HashtableSizeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = HashtableSizeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsHashtableSizeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def HashtableSizeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# HashtableSizeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def HashtableSizeOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def HashtableSizeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class HashtableSizeOptionsT(object):
|
|
|
|
# HashtableSizeOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
hashtableSizeOptions = HashtableSizeOptions()
|
|
hashtableSizeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(hashtableSizeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, hashtableSizeOptions):
|
|
x = HashtableSizeOptionsT()
|
|
x._UnPack(hashtableSizeOptions)
|
|
return x
|
|
|
|
# HashtableSizeOptionsT
|
|
def _UnPack(self, hashtableSizeOptions):
|
|
if hashtableSizeOptions is None:
|
|
return
|
|
|
|
# HashtableSizeOptionsT
|
|
def Pack(self, builder):
|
|
HashtableSizeOptionsStart(builder)
|
|
hashtableSizeOptions = HashtableSizeOptionsEnd(builder)
|
|
return hashtableSizeOptions
|
|
|
|
|
|
class VarHandleOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = VarHandleOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsVarHandleOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def VarHandleOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# VarHandleOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# VarHandleOptions
|
|
def Container(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# VarHandleOptions
|
|
def SharedName(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
def VarHandleOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def VarHandleOptionsAddContainer(builder, container):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(container), 0)
|
|
|
|
def VarHandleOptionsAddSharedName(builder, sharedName):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(sharedName), 0)
|
|
|
|
def VarHandleOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class VarHandleOptionsT(object):
|
|
|
|
# VarHandleOptionsT
|
|
def __init__(self):
|
|
self.container = None # type: str
|
|
self.sharedName = None # type: str
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
varHandleOptions = VarHandleOptions()
|
|
varHandleOptions.Init(buf, pos)
|
|
return cls.InitFromObj(varHandleOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, varHandleOptions):
|
|
x = VarHandleOptionsT()
|
|
x._UnPack(varHandleOptions)
|
|
return x
|
|
|
|
# VarHandleOptionsT
|
|
def _UnPack(self, varHandleOptions):
|
|
if varHandleOptions is None:
|
|
return
|
|
self.container = varHandleOptions.Container()
|
|
self.sharedName = varHandleOptions.SharedName()
|
|
|
|
# VarHandleOptionsT
|
|
def Pack(self, builder):
|
|
if self.container is not None:
|
|
container = builder.CreateString(self.container)
|
|
if self.sharedName is not None:
|
|
sharedName = builder.CreateString(self.sharedName)
|
|
VarHandleOptionsStart(builder)
|
|
if self.container is not None:
|
|
VarHandleOptionsAddContainer(builder, container)
|
|
if self.sharedName is not None:
|
|
VarHandleOptionsAddSharedName(builder, sharedName)
|
|
varHandleOptions = VarHandleOptionsEnd(builder)
|
|
return varHandleOptions
|
|
|
|
|
|
class ReadVariableOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReadVariableOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReadVariableOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReadVariableOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReadVariableOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ReadVariableOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ReadVariableOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ReadVariableOptionsT(object):
|
|
|
|
# ReadVariableOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
readVariableOptions = ReadVariableOptions()
|
|
readVariableOptions.Init(buf, pos)
|
|
return cls.InitFromObj(readVariableOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, readVariableOptions):
|
|
x = ReadVariableOptionsT()
|
|
x._UnPack(readVariableOptions)
|
|
return x
|
|
|
|
# ReadVariableOptionsT
|
|
def _UnPack(self, readVariableOptions):
|
|
if readVariableOptions is None:
|
|
return
|
|
|
|
# ReadVariableOptionsT
|
|
def Pack(self, builder):
|
|
ReadVariableOptionsStart(builder)
|
|
readVariableOptions = ReadVariableOptionsEnd(builder)
|
|
return readVariableOptions
|
|
|
|
|
|
class AssignVariableOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = AssignVariableOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsAssignVariableOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def AssignVariableOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# AssignVariableOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def AssignVariableOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def AssignVariableOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class AssignVariableOptionsT(object):
|
|
|
|
# AssignVariableOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
assignVariableOptions = AssignVariableOptions()
|
|
assignVariableOptions.Init(buf, pos)
|
|
return cls.InitFromObj(assignVariableOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, assignVariableOptions):
|
|
x = AssignVariableOptionsT()
|
|
x._UnPack(assignVariableOptions)
|
|
return x
|
|
|
|
# AssignVariableOptionsT
|
|
def _UnPack(self, assignVariableOptions):
|
|
if assignVariableOptions is None:
|
|
return
|
|
|
|
# AssignVariableOptionsT
|
|
def Pack(self, builder):
|
|
AssignVariableOptionsStart(builder)
|
|
assignVariableOptions = AssignVariableOptionsEnd(builder)
|
|
return assignVariableOptions
|
|
|
|
|
|
class RandomOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = RandomOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRandomOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def RandomOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# RandomOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# RandomOptions
|
|
def Seed(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# RandomOptions
|
|
def Seed2(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def RandomOptionsStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def RandomOptionsAddSeed(builder, seed):
|
|
builder.PrependInt64Slot(0, seed, 0)
|
|
|
|
def RandomOptionsAddSeed2(builder, seed2):
|
|
builder.PrependInt64Slot(1, seed2, 0)
|
|
|
|
def RandomOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class RandomOptionsT(object):
|
|
|
|
# RandomOptionsT
|
|
def __init__(self):
|
|
self.seed = 0 # type: int
|
|
self.seed2 = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
randomOptions = RandomOptions()
|
|
randomOptions.Init(buf, pos)
|
|
return cls.InitFromObj(randomOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, randomOptions):
|
|
x = RandomOptionsT()
|
|
x._UnPack(randomOptions)
|
|
return x
|
|
|
|
# RandomOptionsT
|
|
def _UnPack(self, randomOptions):
|
|
if randomOptions is None:
|
|
return
|
|
self.seed = randomOptions.Seed()
|
|
self.seed2 = randomOptions.Seed2()
|
|
|
|
# RandomOptionsT
|
|
def Pack(self, builder):
|
|
RandomOptionsStart(builder)
|
|
RandomOptionsAddSeed(builder, self.seed)
|
|
RandomOptionsAddSeed2(builder, self.seed2)
|
|
randomOptions = RandomOptionsEnd(builder)
|
|
return randomOptions
|
|
|
|
|
|
class BucketizeOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BucketizeOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBucketizeOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BucketizeOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BucketizeOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# BucketizeOptions
|
|
def Boundaries(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.Float32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# BucketizeOptions
|
|
def BoundariesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Float32Flags, o)
|
|
return 0
|
|
|
|
# BucketizeOptions
|
|
def BoundariesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# BucketizeOptions
|
|
def BoundariesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
def BucketizeOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def BucketizeOptionsAddBoundaries(builder, boundaries):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(boundaries), 0)
|
|
|
|
def BucketizeOptionsStartBoundariesVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def BucketizeOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class BucketizeOptionsT(object):
|
|
|
|
# BucketizeOptionsT
|
|
def __init__(self):
|
|
self.boundaries = None # type: List[float]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
bucketizeOptions = BucketizeOptions()
|
|
bucketizeOptions.Init(buf, pos)
|
|
return cls.InitFromObj(bucketizeOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, bucketizeOptions):
|
|
x = BucketizeOptionsT()
|
|
x._UnPack(bucketizeOptions)
|
|
return x
|
|
|
|
# BucketizeOptionsT
|
|
def _UnPack(self, bucketizeOptions):
|
|
if bucketizeOptions is None:
|
|
return
|
|
if not bucketizeOptions.BoundariesIsNone():
|
|
if np is None:
|
|
self.boundaries = []
|
|
for i in range(bucketizeOptions.BoundariesLength()):
|
|
self.boundaries.append(bucketizeOptions.Boundaries(i))
|
|
else:
|
|
self.boundaries = bucketizeOptions.BoundariesAsNumpy()
|
|
|
|
# BucketizeOptionsT
|
|
def Pack(self, builder):
|
|
if self.boundaries is not None:
|
|
if np is not None and type(self.boundaries) is np.ndarray:
|
|
boundaries = builder.CreateNumpyVector(self.boundaries)
|
|
else:
|
|
BucketizeOptionsStartBoundariesVector(builder, len(self.boundaries))
|
|
for i in reversed(range(len(self.boundaries))):
|
|
builder.PrependFloat32(self.boundaries[i])
|
|
boundaries = builder.EndVector()
|
|
BucketizeOptionsStart(builder)
|
|
if self.boundaries is not None:
|
|
BucketizeOptionsAddBoundaries(builder, boundaries)
|
|
bucketizeOptions = BucketizeOptionsEnd(builder)
|
|
return bucketizeOptions
|
|
|
|
|
|
class GeluOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = GeluOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsGeluOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def GeluOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# GeluOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# GeluOptions
|
|
def Approximate(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
|
|
return False
|
|
|
|
def GeluOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def GeluOptionsAddApproximate(builder, approximate):
|
|
builder.PrependBoolSlot(0, approximate, 0)
|
|
|
|
def GeluOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class GeluOptionsT(object):
|
|
|
|
# GeluOptionsT
|
|
def __init__(self):
|
|
self.approximate = False # type: bool
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
geluOptions = GeluOptions()
|
|
geluOptions.Init(buf, pos)
|
|
return cls.InitFromObj(geluOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, geluOptions):
|
|
x = GeluOptionsT()
|
|
x._UnPack(geluOptions)
|
|
return x
|
|
|
|
# GeluOptionsT
|
|
def _UnPack(self, geluOptions):
|
|
if geluOptions is None:
|
|
return
|
|
self.approximate = geluOptions.Approximate()
|
|
|
|
# GeluOptionsT
|
|
def Pack(self, builder):
|
|
GeluOptionsStart(builder)
|
|
GeluOptionsAddApproximate(builder, self.approximate)
|
|
geluOptions = GeluOptionsEnd(builder)
|
|
return geluOptions
|
|
|
|
|
|
class DynamicUpdateSliceOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DynamicUpdateSliceOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDynamicUpdateSliceOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DynamicUpdateSliceOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DynamicUpdateSliceOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def DynamicUpdateSliceOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def DynamicUpdateSliceOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DynamicUpdateSliceOptionsT(object):
|
|
|
|
# DynamicUpdateSliceOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
dynamicUpdateSliceOptions = DynamicUpdateSliceOptions()
|
|
dynamicUpdateSliceOptions.Init(buf, pos)
|
|
return cls.InitFromObj(dynamicUpdateSliceOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, dynamicUpdateSliceOptions):
|
|
x = DynamicUpdateSliceOptionsT()
|
|
x._UnPack(dynamicUpdateSliceOptions)
|
|
return x
|
|
|
|
# DynamicUpdateSliceOptionsT
|
|
def _UnPack(self, dynamicUpdateSliceOptions):
|
|
if dynamicUpdateSliceOptions is None:
|
|
return
|
|
|
|
# DynamicUpdateSliceOptionsT
|
|
def Pack(self, builder):
|
|
DynamicUpdateSliceOptionsStart(builder)
|
|
dynamicUpdateSliceOptions = DynamicUpdateSliceOptionsEnd(builder)
|
|
return dynamicUpdateSliceOptions
|
|
|
|
|
|
class UnsortedSegmentProdOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnsortedSegmentProdOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnsortedSegmentProdOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnsortedSegmentProdOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnsortedSegmentProdOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def UnsortedSegmentProdOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def UnsortedSegmentProdOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnsortedSegmentProdOptionsT(object):
|
|
|
|
# UnsortedSegmentProdOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unsortedSegmentProdOptions = UnsortedSegmentProdOptions()
|
|
unsortedSegmentProdOptions.Init(buf, pos)
|
|
return cls.InitFromObj(unsortedSegmentProdOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unsortedSegmentProdOptions):
|
|
x = UnsortedSegmentProdOptionsT()
|
|
x._UnPack(unsortedSegmentProdOptions)
|
|
return x
|
|
|
|
# UnsortedSegmentProdOptionsT
|
|
def _UnPack(self, unsortedSegmentProdOptions):
|
|
if unsortedSegmentProdOptions is None:
|
|
return
|
|
|
|
# UnsortedSegmentProdOptionsT
|
|
def Pack(self, builder):
|
|
UnsortedSegmentProdOptionsStart(builder)
|
|
unsortedSegmentProdOptions = UnsortedSegmentProdOptionsEnd(builder)
|
|
return unsortedSegmentProdOptions
|
|
|
|
|
|
class UnsortedSegmentMaxOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnsortedSegmentMaxOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnsortedSegmentMaxOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnsortedSegmentMaxOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnsortedSegmentMaxOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def UnsortedSegmentMaxOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def UnsortedSegmentMaxOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnsortedSegmentMaxOptionsT(object):
|
|
|
|
# UnsortedSegmentMaxOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unsortedSegmentMaxOptions = UnsortedSegmentMaxOptions()
|
|
unsortedSegmentMaxOptions.Init(buf, pos)
|
|
return cls.InitFromObj(unsortedSegmentMaxOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unsortedSegmentMaxOptions):
|
|
x = UnsortedSegmentMaxOptionsT()
|
|
x._UnPack(unsortedSegmentMaxOptions)
|
|
return x
|
|
|
|
# UnsortedSegmentMaxOptionsT
|
|
def _UnPack(self, unsortedSegmentMaxOptions):
|
|
if unsortedSegmentMaxOptions is None:
|
|
return
|
|
|
|
# UnsortedSegmentMaxOptionsT
|
|
def Pack(self, builder):
|
|
UnsortedSegmentMaxOptionsStart(builder)
|
|
unsortedSegmentMaxOptions = UnsortedSegmentMaxOptionsEnd(builder)
|
|
return unsortedSegmentMaxOptions
|
|
|
|
|
|
class UnsortedSegmentSumOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnsortedSegmentSumOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnsortedSegmentSumOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnsortedSegmentSumOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnsortedSegmentSumOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def UnsortedSegmentSumOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def UnsortedSegmentSumOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnsortedSegmentSumOptionsT(object):
|
|
|
|
# UnsortedSegmentSumOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unsortedSegmentSumOptions = UnsortedSegmentSumOptions()
|
|
unsortedSegmentSumOptions.Init(buf, pos)
|
|
return cls.InitFromObj(unsortedSegmentSumOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unsortedSegmentSumOptions):
|
|
x = UnsortedSegmentSumOptionsT()
|
|
x._UnPack(unsortedSegmentSumOptions)
|
|
return x
|
|
|
|
# UnsortedSegmentSumOptionsT
|
|
def _UnPack(self, unsortedSegmentSumOptions):
|
|
if unsortedSegmentSumOptions is None:
|
|
return
|
|
|
|
# UnsortedSegmentSumOptionsT
|
|
def Pack(self, builder):
|
|
UnsortedSegmentSumOptionsStart(builder)
|
|
unsortedSegmentSumOptions = UnsortedSegmentSumOptionsEnd(builder)
|
|
return unsortedSegmentSumOptions
|
|
|
|
|
|
class ATan2Options(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ATan2Options()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsATan2Options(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ATan2OptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ATan2Options
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def ATan2OptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def ATan2OptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ATan2OptionsT(object):
|
|
|
|
# ATan2OptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
atan2Options = ATan2Options()
|
|
atan2Options.Init(buf, pos)
|
|
return cls.InitFromObj(atan2Options)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, atan2Options):
|
|
x = ATan2OptionsT()
|
|
x._UnPack(atan2Options)
|
|
return x
|
|
|
|
# ATan2OptionsT
|
|
def _UnPack(self, atan2Options):
|
|
if atan2Options is None:
|
|
return
|
|
|
|
# ATan2OptionsT
|
|
def Pack(self, builder):
|
|
ATan2OptionsStart(builder)
|
|
atan2Options = ATan2OptionsEnd(builder)
|
|
return atan2Options
|
|
|
|
|
|
class UnsortedSegmentMinOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = UnsortedSegmentMinOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsUnsortedSegmentMinOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def UnsortedSegmentMinOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# UnsortedSegmentMinOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def UnsortedSegmentMinOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def UnsortedSegmentMinOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class UnsortedSegmentMinOptionsT(object):
|
|
|
|
# UnsortedSegmentMinOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
unsortedSegmentMinOptions = UnsortedSegmentMinOptions()
|
|
unsortedSegmentMinOptions.Init(buf, pos)
|
|
return cls.InitFromObj(unsortedSegmentMinOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, unsortedSegmentMinOptions):
|
|
x = UnsortedSegmentMinOptionsT()
|
|
x._UnPack(unsortedSegmentMinOptions)
|
|
return x
|
|
|
|
# UnsortedSegmentMinOptionsT
|
|
def _UnPack(self, unsortedSegmentMinOptions):
|
|
if unsortedSegmentMinOptions is None:
|
|
return
|
|
|
|
# UnsortedSegmentMinOptionsT
|
|
def Pack(self, builder):
|
|
UnsortedSegmentMinOptionsStart(builder)
|
|
unsortedSegmentMinOptions = UnsortedSegmentMinOptionsEnd(builder)
|
|
return unsortedSegmentMinOptions
|
|
|
|
|
|
class SignOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SignOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSignOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SignOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SignOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def SignOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def SignOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class SignOptionsT(object):
|
|
|
|
# SignOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
signOptions = SignOptions()
|
|
signOptions.Init(buf, pos)
|
|
return cls.InitFromObj(signOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, signOptions):
|
|
x = SignOptionsT()
|
|
x._UnPack(signOptions)
|
|
return x
|
|
|
|
# SignOptionsT
|
|
def _UnPack(self, signOptions):
|
|
if signOptions is None:
|
|
return
|
|
|
|
# SignOptionsT
|
|
def Pack(self, builder):
|
|
SignOptionsStart(builder)
|
|
signOptions = SignOptionsEnd(builder)
|
|
return signOptions
|
|
|
|
|
|
class BitcastOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BitcastOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBitcastOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BitcastOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BitcastOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def BitcastOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def BitcastOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BitcastOptionsT(object):
|
|
|
|
# BitcastOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
bitcastOptions = BitcastOptions()
|
|
bitcastOptions.Init(buf, pos)
|
|
return cls.InitFromObj(bitcastOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, bitcastOptions):
|
|
x = BitcastOptionsT()
|
|
x._UnPack(bitcastOptions)
|
|
return x
|
|
|
|
# BitcastOptionsT
|
|
def _UnPack(self, bitcastOptions):
|
|
if bitcastOptions is None:
|
|
return
|
|
|
|
# BitcastOptionsT
|
|
def Pack(self, builder):
|
|
BitcastOptionsStart(builder)
|
|
bitcastOptions = BitcastOptionsEnd(builder)
|
|
return bitcastOptions
|
|
|
|
|
|
class BitwiseXorOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = BitwiseXorOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBitwiseXorOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BitwiseXorOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# BitwiseXorOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def BitwiseXorOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def BitwiseXorOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class BitwiseXorOptionsT(object):
|
|
|
|
# BitwiseXorOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
bitwiseXorOptions = BitwiseXorOptions()
|
|
bitwiseXorOptions.Init(buf, pos)
|
|
return cls.InitFromObj(bitwiseXorOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, bitwiseXorOptions):
|
|
x = BitwiseXorOptionsT()
|
|
x._UnPack(bitwiseXorOptions)
|
|
return x
|
|
|
|
# BitwiseXorOptionsT
|
|
def _UnPack(self, bitwiseXorOptions):
|
|
if bitwiseXorOptions is None:
|
|
return
|
|
|
|
# BitwiseXorOptionsT
|
|
def Pack(self, builder):
|
|
BitwiseXorOptionsStart(builder)
|
|
bitwiseXorOptions = BitwiseXorOptionsEnd(builder)
|
|
return bitwiseXorOptions
|
|
|
|
|
|
class RightShiftOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = RightShiftOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsRightShiftOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def RightShiftOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# RightShiftOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def RightShiftOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def RightShiftOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class RightShiftOptionsT(object):
|
|
|
|
# RightShiftOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
rightShiftOptions = RightShiftOptions()
|
|
rightShiftOptions.Init(buf, pos)
|
|
return cls.InitFromObj(rightShiftOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, rightShiftOptions):
|
|
x = RightShiftOptionsT()
|
|
x._UnPack(rightShiftOptions)
|
|
return x
|
|
|
|
# RightShiftOptionsT
|
|
def _UnPack(self, rightShiftOptions):
|
|
if rightShiftOptions is None:
|
|
return
|
|
|
|
# RightShiftOptionsT
|
|
def Pack(self, builder):
|
|
RightShiftOptionsStart(builder)
|
|
rightShiftOptions = RightShiftOptionsEnd(builder)
|
|
return rightShiftOptions
|
|
|
|
|
|
class DilateOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = DilateOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsDilateOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def DilateOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# DilateOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
def DilateOptionsStart(builder):
|
|
builder.StartObject(0)
|
|
|
|
def DilateOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class DilateOptionsT(object):
|
|
|
|
# DilateOptionsT
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
dilateOptions = DilateOptions()
|
|
dilateOptions.Init(buf, pos)
|
|
return cls.InitFromObj(dilateOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, dilateOptions):
|
|
x = DilateOptionsT()
|
|
x._UnPack(dilateOptions)
|
|
return x
|
|
|
|
# DilateOptionsT
|
|
def _UnPack(self, dilateOptions):
|
|
if dilateOptions is None:
|
|
return
|
|
|
|
# DilateOptionsT
|
|
def Pack(self, builder):
|
|
DilateOptionsStart(builder)
|
|
dilateOptions = DilateOptionsEnd(builder)
|
|
return dilateOptions
|
|
|
|
|
|
class ReduceWindowOptions(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = ReduceWindowOptions()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsReduceWindowOptions(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ReduceWindowOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# ReduceWindowOptions
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# ReduceWindowOptions
|
|
def ReduceFunction(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def ReduceWindowOptionsStart(builder):
|
|
builder.StartObject(1)
|
|
|
|
def ReduceWindowOptionsAddReduceFunction(builder, reduceFunction):
|
|
builder.PrependInt32Slot(0, reduceFunction, 0)
|
|
|
|
def ReduceWindowOptionsEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class ReduceWindowOptionsT(object):
|
|
|
|
# ReduceWindowOptionsT
|
|
def __init__(self):
|
|
self.reduceFunction = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
reduceWindowOptions = ReduceWindowOptions()
|
|
reduceWindowOptions.Init(buf, pos)
|
|
return cls.InitFromObj(reduceWindowOptions)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, reduceWindowOptions):
|
|
x = ReduceWindowOptionsT()
|
|
x._UnPack(reduceWindowOptions)
|
|
return x
|
|
|
|
# ReduceWindowOptionsT
|
|
def _UnPack(self, reduceWindowOptions):
|
|
if reduceWindowOptions is None:
|
|
return
|
|
self.reduceFunction = reduceWindowOptions.ReduceFunction()
|
|
|
|
# ReduceWindowOptionsT
|
|
def Pack(self, builder):
|
|
ReduceWindowOptionsStart(builder)
|
|
ReduceWindowOptionsAddReduceFunction(builder, self.reduceFunction)
|
|
reduceWindowOptions = ReduceWindowOptionsEnd(builder)
|
|
return reduceWindowOptions
|
|
|
|
|
|
class OperatorCode(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = OperatorCode()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsOperatorCode(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def OperatorCodeBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# OperatorCode
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# OperatorCode
|
|
def DeprecatedBuiltinCode(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# OperatorCode
|
|
def CustomCode(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# OperatorCode
|
|
def Version(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 1
|
|
|
|
# OperatorCode
|
|
def BuiltinCode(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def OperatorCodeStart(builder):
|
|
builder.StartObject(4)
|
|
|
|
def OperatorCodeAddDeprecatedBuiltinCode(builder, deprecatedBuiltinCode):
|
|
builder.PrependInt8Slot(0, deprecatedBuiltinCode, 0)
|
|
|
|
def OperatorCodeAddCustomCode(builder, customCode):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(customCode), 0)
|
|
|
|
def OperatorCodeAddVersion(builder, version):
|
|
builder.PrependInt32Slot(2, version, 1)
|
|
|
|
def OperatorCodeAddBuiltinCode(builder, builtinCode):
|
|
builder.PrependInt32Slot(3, builtinCode, 0)
|
|
|
|
def OperatorCodeEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class OperatorCodeT(object):
|
|
|
|
# OperatorCodeT
|
|
def __init__(self):
|
|
self.deprecatedBuiltinCode = 0 # type: int
|
|
self.customCode = None # type: str
|
|
self.version = 1 # type: int
|
|
self.builtinCode = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
operatorCode = OperatorCode()
|
|
operatorCode.Init(buf, pos)
|
|
return cls.InitFromObj(operatorCode)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, operatorCode):
|
|
x = OperatorCodeT()
|
|
x._UnPack(operatorCode)
|
|
return x
|
|
|
|
# OperatorCodeT
|
|
def _UnPack(self, operatorCode):
|
|
if operatorCode is None:
|
|
return
|
|
self.deprecatedBuiltinCode = operatorCode.DeprecatedBuiltinCode()
|
|
self.customCode = operatorCode.CustomCode()
|
|
self.version = operatorCode.Version()
|
|
self.builtinCode = operatorCode.BuiltinCode()
|
|
|
|
# OperatorCodeT
|
|
def Pack(self, builder):
|
|
if self.customCode is not None:
|
|
customCode = builder.CreateString(self.customCode)
|
|
OperatorCodeStart(builder)
|
|
OperatorCodeAddDeprecatedBuiltinCode(builder, self.deprecatedBuiltinCode)
|
|
if self.customCode is not None:
|
|
OperatorCodeAddCustomCode(builder, customCode)
|
|
OperatorCodeAddVersion(builder, self.version)
|
|
OperatorCodeAddBuiltinCode(builder, self.builtinCode)
|
|
operatorCode = OperatorCodeEnd(builder)
|
|
return operatorCode
|
|
|
|
|
|
class Operator(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Operator()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsOperator(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def OperatorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Operator
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Operator
|
|
def OpcodeIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def Inputs(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Operator
|
|
def InputsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Operator
|
|
def InputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Operator
|
|
def InputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# Operator
|
|
def Outputs(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Operator
|
|
def OutputsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Operator
|
|
def OutputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Operator
|
|
def OutputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# Operator
|
|
def BuiltinOptionsType(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def BuiltinOptions(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
from flatbuffers.table import Table
|
|
obj = Table(bytearray(), 0)
|
|
self._tab.Union(obj, o)
|
|
return obj
|
|
return None
|
|
|
|
# Operator
|
|
def CustomOptions(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
a = self._tab.Vector(o)
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# Operator
|
|
def CustomOptionsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o)
|
|
return 0
|
|
|
|
# Operator
|
|
def CustomOptionsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Operator
|
|
def CustomOptionsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
return o == 0
|
|
|
|
# Operator
|
|
def CustomOptionsFormat(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def MutatingVariableInputs(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.BoolFlags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# Operator
|
|
def MutatingVariableInputsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.BoolFlags, o)
|
|
return 0
|
|
|
|
# Operator
|
|
def MutatingVariableInputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Operator
|
|
def MutatingVariableInputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
return o == 0
|
|
|
|
# Operator
|
|
def Intermediates(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
if o != 0:
|
|
a = self._tab.Vector(o)
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Operator
|
|
def IntermediatesAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Operator
|
|
def IntermediatesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Operator
|
|
def IntermediatesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
return o == 0
|
|
|
|
# Operator
|
|
def LargeCustomOptionsOffset(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def LargeCustomOptionsSize(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def BuiltinOptions2Type(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint8Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Operator
|
|
def BuiltinOptions2(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(28))
|
|
if o != 0:
|
|
from flatbuffers.table import Table
|
|
obj = Table(bytearray(), 0)
|
|
self._tab.Union(obj, o)
|
|
return obj
|
|
return None
|
|
|
|
def OperatorStart(builder):
|
|
builder.StartObject(13)
|
|
|
|
def OperatorAddOpcodeIndex(builder, opcodeIndex):
|
|
builder.PrependUint32Slot(0, opcodeIndex, 0)
|
|
|
|
def OperatorAddInputs(builder, inputs):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(inputs), 0)
|
|
|
|
def OperatorStartInputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def OperatorAddOutputs(builder, outputs):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(outputs), 0)
|
|
|
|
def OperatorStartOutputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def OperatorAddBuiltinOptionsType(builder, builtinOptionsType):
|
|
builder.PrependUint8Slot(3, builtinOptionsType, 0)
|
|
|
|
def OperatorAddBuiltinOptions(builder, builtinOptions):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(builtinOptions), 0)
|
|
|
|
def OperatorAddCustomOptions(builder, customOptions):
|
|
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(customOptions), 0)
|
|
|
|
def OperatorStartCustomOptionsVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def OperatorAddCustomOptionsFormat(builder, customOptionsFormat):
|
|
builder.PrependInt8Slot(6, customOptionsFormat, 0)
|
|
|
|
def OperatorAddMutatingVariableInputs(builder, mutatingVariableInputs):
|
|
builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(mutatingVariableInputs), 0)
|
|
|
|
def OperatorStartMutatingVariableInputsVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def OperatorAddIntermediates(builder, intermediates):
|
|
builder.PrependUOffsetTRelativeSlot(8, flatbuffers.number_types.UOffsetTFlags.py_type(intermediates), 0)
|
|
|
|
def OperatorStartIntermediatesVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def OperatorAddLargeCustomOptionsOffset(builder, largeCustomOptionsOffset):
|
|
builder.PrependUint64Slot(9, largeCustomOptionsOffset, 0)
|
|
|
|
def OperatorAddLargeCustomOptionsSize(builder, largeCustomOptionsSize):
|
|
builder.PrependUint64Slot(10, largeCustomOptionsSize, 0)
|
|
|
|
def OperatorAddBuiltinOptions2Type(builder, builtinOptions2Type):
|
|
builder.PrependUint8Slot(11, builtinOptions2Type, 0)
|
|
|
|
def OperatorAddBuiltinOptions2(builder, builtinOptions2):
|
|
builder.PrependUOffsetTRelativeSlot(12, flatbuffers.number_types.UOffsetTFlags.py_type(builtinOptions2), 0)
|
|
|
|
def OperatorEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List, Union
|
|
except:
|
|
pass
|
|
|
|
class OperatorT(object):
|
|
|
|
# OperatorT
|
|
def __init__(self):
|
|
self.opcodeIndex = 0 # type: int
|
|
self.inputs = None # type: List[int]
|
|
self.outputs = None # type: List[int]
|
|
self.builtinOptionsType = 0 # type: int
|
|
self.builtinOptions = None # type: Union[None, Conv2DOptionsT, DepthwiseConv2DOptionsT, ConcatEmbeddingsOptionsT, LSHProjectionOptionsT, Pool2DOptionsT, SVDFOptionsT, RNNOptionsT, FullyConnectedOptionsT, SoftmaxOptionsT, ConcatenationOptionsT, AddOptionsT, L2NormOptionsT, LocalResponseNormalizationOptionsT, LSTMOptionsT, ResizeBilinearOptionsT, CallOptionsT, ReshapeOptionsT, SkipGramOptionsT, SpaceToDepthOptionsT, EmbeddingLookupSparseOptionsT, MulOptionsT, PadOptionsT, GatherOptionsT, BatchToSpaceNDOptionsT, SpaceToBatchNDOptionsT, TransposeOptionsT, ReducerOptionsT, SubOptionsT, DivOptionsT, SqueezeOptionsT, SequenceRNNOptionsT, StridedSliceOptionsT, ExpOptionsT, TopKV2OptionsT, SplitOptionsT, LogSoftmaxOptionsT, CastOptionsT, DequantizeOptionsT, MaximumMinimumOptionsT, ArgMaxOptionsT, LessOptionsT, NegOptionsT, PadV2OptionsT, GreaterOptionsT, GreaterEqualOptionsT, LessEqualOptionsT, SelectOptionsT, SliceOptionsT, TransposeConvOptionsT, SparseToDenseOptionsT, TileOptionsT, ExpandDimsOptionsT, EqualOptionsT, NotEqualOptionsT, ShapeOptionsT, PowOptionsT, ArgMinOptionsT, FakeQuantOptionsT, PackOptionsT, LogicalOrOptionsT, OneHotOptionsT, LogicalAndOptionsT, LogicalNotOptionsT, UnpackOptionsT, FloorDivOptionsT, SquareOptionsT, ZerosLikeOptionsT, FillOptionsT, BidirectionalSequenceLSTMOptionsT, BidirectionalSequenceRNNOptionsT, UnidirectionalSequenceLSTMOptionsT, FloorModOptionsT, RangeOptionsT, ResizeNearestNeighborOptionsT, LeakyReluOptionsT, SquaredDifferenceOptionsT, MirrorPadOptionsT, AbsOptionsT, SplitVOptionsT, UniqueOptionsT, ReverseV2OptionsT, AddNOptionsT, GatherNdOptionsT, CosOptionsT, WhereOptionsT, RankOptionsT, ReverseSequenceOptionsT, MatrixDiagOptionsT, QuantizeOptionsT, MatrixSetDiagOptionsT, HardSwishOptionsT, IfOptionsT, WhileOptionsT, DepthToSpaceOptionsT, NonMaxSuppressionV4OptionsT, NonMaxSuppressionV5OptionsT, ScatterNdOptionsT, SelectV2OptionsT, DensifyOptionsT, SegmentSumOptionsT, BatchMatMulOptionsT, CumsumOptionsT, CallOnceOptionsT, BroadcastToOptionsT, Rfft2dOptionsT, Conv3DOptionsT, HashtableOptionsT, HashtableFindOptionsT, HashtableImportOptionsT, HashtableSizeOptionsT, VarHandleOptionsT, ReadVariableOptionsT, AssignVariableOptionsT, RandomOptionsT, BucketizeOptionsT, GeluOptionsT, DynamicUpdateSliceOptionsT, UnsortedSegmentProdOptionsT, UnsortedSegmentMaxOptionsT, UnsortedSegmentMinOptionsT, UnsortedSegmentSumOptionsT, ATan2OptionsT, SignOptionsT, BitcastOptionsT, BitwiseXorOptionsT, RightShiftOptionsT]
|
|
self.customOptions = None # type: List[int]
|
|
self.customOptionsFormat = 0 # type: int
|
|
self.mutatingVariableInputs = None # type: List[bool]
|
|
self.intermediates = None # type: List[int]
|
|
self.largeCustomOptionsOffset = 0 # type: int
|
|
self.largeCustomOptionsSize = 0 # type: int
|
|
self.builtinOptions2Type = 0 # type: int
|
|
self.builtinOptions2 = None # type: Union[None, StablehloConcatenateOptionsT, StablehloBroadcastInDimOptionsT, StablehloSliceOptionsT, StablehloConvolutionOptionsT, StablehloCustomCallOptionsT, StablehloReduceOptionsT, StablehloScatterOptionsT, StablehloCompareOptionsT, StablehloDynamicSliceOptionsT, StablehloPadOptionsT, StablehloIotaOptionsT, StablehloDotGeneralOptionsT, StablehloReduceWindowOptionsT, StablehloSortOptionsT, StablehloWhileOptionsT, StablehloGatherOptionsT, StablehloTransposeOptionsT, DilateOptionsT, StablehloRngBitGeneratorOptionsT, ReduceWindowOptionsT]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
operator = Operator()
|
|
operator.Init(buf, pos)
|
|
return cls.InitFromObj(operator)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, operator):
|
|
x = OperatorT()
|
|
x._UnPack(operator)
|
|
return x
|
|
|
|
# OperatorT
|
|
def _UnPack(self, operator):
|
|
if operator is None:
|
|
return
|
|
self.opcodeIndex = operator.OpcodeIndex()
|
|
if not operator.InputsIsNone():
|
|
if np is None:
|
|
self.inputs = []
|
|
for i in range(operator.InputsLength()):
|
|
self.inputs.append(operator.Inputs(i))
|
|
else:
|
|
self.inputs = operator.InputsAsNumpy()
|
|
if not operator.OutputsIsNone():
|
|
if np is None:
|
|
self.outputs = []
|
|
for i in range(operator.OutputsLength()):
|
|
self.outputs.append(operator.Outputs(i))
|
|
else:
|
|
self.outputs = operator.OutputsAsNumpy()
|
|
self.builtinOptionsType = operator.BuiltinOptionsType()
|
|
self.builtinOptions = BuiltinOptionsCreator(self.builtinOptionsType, operator.BuiltinOptions())
|
|
if not operator.CustomOptionsIsNone():
|
|
if np is None:
|
|
self.customOptions = []
|
|
for i in range(operator.CustomOptionsLength()):
|
|
self.customOptions.append(operator.CustomOptions(i))
|
|
else:
|
|
self.customOptions = operator.CustomOptionsAsNumpy()
|
|
self.customOptionsFormat = operator.CustomOptionsFormat()
|
|
if not operator.MutatingVariableInputsIsNone():
|
|
if np is None:
|
|
self.mutatingVariableInputs = []
|
|
for i in range(operator.MutatingVariableInputsLength()):
|
|
self.mutatingVariableInputs.append(operator.MutatingVariableInputs(i))
|
|
else:
|
|
self.mutatingVariableInputs = operator.MutatingVariableInputsAsNumpy()
|
|
if not operator.IntermediatesIsNone():
|
|
if np is None:
|
|
self.intermediates = []
|
|
for i in range(operator.IntermediatesLength()):
|
|
self.intermediates.append(operator.Intermediates(i))
|
|
else:
|
|
self.intermediates = operator.IntermediatesAsNumpy()
|
|
self.largeCustomOptionsOffset = operator.LargeCustomOptionsOffset()
|
|
self.largeCustomOptionsSize = operator.LargeCustomOptionsSize()
|
|
self.builtinOptions2Type = operator.BuiltinOptions2Type()
|
|
self.builtinOptions2 = BuiltinOptions2Creator(self.builtinOptions2Type, operator.BuiltinOptions2())
|
|
|
|
# OperatorT
|
|
def Pack(self, builder):
|
|
if self.inputs is not None:
|
|
if np is not None and type(self.inputs) is np.ndarray:
|
|
inputs = builder.CreateNumpyVector(self.inputs)
|
|
else:
|
|
OperatorStartInputsVector(builder, len(self.inputs))
|
|
for i in reversed(range(len(self.inputs))):
|
|
builder.PrependInt32(self.inputs[i])
|
|
inputs = builder.EndVector()
|
|
if self.outputs is not None:
|
|
if np is not None and type(self.outputs) is np.ndarray:
|
|
outputs = builder.CreateNumpyVector(self.outputs)
|
|
else:
|
|
OperatorStartOutputsVector(builder, len(self.outputs))
|
|
for i in reversed(range(len(self.outputs))):
|
|
builder.PrependInt32(self.outputs[i])
|
|
outputs = builder.EndVector()
|
|
if self.builtinOptions is not None:
|
|
builtinOptions = self.builtinOptions.Pack(builder)
|
|
if self.customOptions is not None:
|
|
if np is not None and type(self.customOptions) is np.ndarray:
|
|
customOptions = builder.CreateNumpyVector(self.customOptions)
|
|
else:
|
|
OperatorStartCustomOptionsVector(builder, len(self.customOptions))
|
|
for i in reversed(range(len(self.customOptions))):
|
|
builder.PrependUint8(self.customOptions[i])
|
|
customOptions = builder.EndVector()
|
|
if self.mutatingVariableInputs is not None:
|
|
if np is not None and type(self.mutatingVariableInputs) is np.ndarray:
|
|
mutatingVariableInputs = builder.CreateNumpyVector(self.mutatingVariableInputs)
|
|
else:
|
|
OperatorStartMutatingVariableInputsVector(builder, len(self.mutatingVariableInputs))
|
|
for i in reversed(range(len(self.mutatingVariableInputs))):
|
|
builder.PrependBool(self.mutatingVariableInputs[i])
|
|
mutatingVariableInputs = builder.EndVector()
|
|
if self.intermediates is not None:
|
|
if np is not None and type(self.intermediates) is np.ndarray:
|
|
intermediates = builder.CreateNumpyVector(self.intermediates)
|
|
else:
|
|
OperatorStartIntermediatesVector(builder, len(self.intermediates))
|
|
for i in reversed(range(len(self.intermediates))):
|
|
builder.PrependInt32(self.intermediates[i])
|
|
intermediates = builder.EndVector()
|
|
if self.builtinOptions2 is not None:
|
|
builtinOptions2 = self.builtinOptions2.Pack(builder)
|
|
OperatorStart(builder)
|
|
OperatorAddOpcodeIndex(builder, self.opcodeIndex)
|
|
if self.inputs is not None:
|
|
OperatorAddInputs(builder, inputs)
|
|
if self.outputs is not None:
|
|
OperatorAddOutputs(builder, outputs)
|
|
OperatorAddBuiltinOptionsType(builder, self.builtinOptionsType)
|
|
if self.builtinOptions is not None:
|
|
OperatorAddBuiltinOptions(builder, builtinOptions)
|
|
if self.customOptions is not None:
|
|
OperatorAddCustomOptions(builder, customOptions)
|
|
OperatorAddCustomOptionsFormat(builder, self.customOptionsFormat)
|
|
if self.mutatingVariableInputs is not None:
|
|
OperatorAddMutatingVariableInputs(builder, mutatingVariableInputs)
|
|
if self.intermediates is not None:
|
|
OperatorAddIntermediates(builder, intermediates)
|
|
OperatorAddLargeCustomOptionsOffset(builder, self.largeCustomOptionsOffset)
|
|
OperatorAddLargeCustomOptionsSize(builder, self.largeCustomOptionsSize)
|
|
OperatorAddBuiltinOptions2Type(builder, self.builtinOptions2Type)
|
|
if self.builtinOptions2 is not None:
|
|
OperatorAddBuiltinOptions2(builder, builtinOptions2)
|
|
operator = OperatorEnd(builder)
|
|
return operator
|
|
|
|
|
|
class SubGraph(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SubGraph()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSubGraph(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SubGraphBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SubGraph
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SubGraph
|
|
def Tensors(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = Tensor()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# SubGraph
|
|
def TensorsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def TensorsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# SubGraph
|
|
def Inputs(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# SubGraph
|
|
def InputsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def InputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def InputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# SubGraph
|
|
def Outputs(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.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# SubGraph
|
|
def OutputsAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def OutputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def OutputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# SubGraph
|
|
def Operators(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = Operator()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# SubGraph
|
|
def OperatorsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SubGraph
|
|
def OperatorsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
return o == 0
|
|
|
|
# SubGraph
|
|
def Name(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
def SubGraphStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def SubGraphAddTensors(builder, tensors):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(tensors), 0)
|
|
|
|
def SubGraphStartTensorsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SubGraphAddInputs(builder, inputs):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(inputs), 0)
|
|
|
|
def SubGraphStartInputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SubGraphAddOutputs(builder, outputs):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(outputs), 0)
|
|
|
|
def SubGraphStartOutputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SubGraphAddOperators(builder, operators):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(operators), 0)
|
|
|
|
def SubGraphStartOperatorsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SubGraphAddName(builder, name):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
|
|
|
|
def SubGraphEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class SubGraphT(object):
|
|
|
|
# SubGraphT
|
|
def __init__(self):
|
|
self.tensors = None # type: List[TensorT]
|
|
self.inputs = None # type: List[int]
|
|
self.outputs = None # type: List[int]
|
|
self.operators = None # type: List[OperatorT]
|
|
self.name = None # type: str
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
subGraph = SubGraph()
|
|
subGraph.Init(buf, pos)
|
|
return cls.InitFromObj(subGraph)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, subGraph):
|
|
x = SubGraphT()
|
|
x._UnPack(subGraph)
|
|
return x
|
|
|
|
# SubGraphT
|
|
def _UnPack(self, subGraph):
|
|
if subGraph is None:
|
|
return
|
|
if not subGraph.TensorsIsNone():
|
|
self.tensors = []
|
|
for i in range(subGraph.TensorsLength()):
|
|
if subGraph.Tensors(i) is None:
|
|
self.tensors.append(None)
|
|
else:
|
|
tensor_ = TensorT.InitFromObj(subGraph.Tensors(i))
|
|
self.tensors.append(tensor_)
|
|
if not subGraph.InputsIsNone():
|
|
if np is None:
|
|
self.inputs = []
|
|
for i in range(subGraph.InputsLength()):
|
|
self.inputs.append(subGraph.Inputs(i))
|
|
else:
|
|
self.inputs = subGraph.InputsAsNumpy()
|
|
if not subGraph.OutputsIsNone():
|
|
if np is None:
|
|
self.outputs = []
|
|
for i in range(subGraph.OutputsLength()):
|
|
self.outputs.append(subGraph.Outputs(i))
|
|
else:
|
|
self.outputs = subGraph.OutputsAsNumpy()
|
|
if not subGraph.OperatorsIsNone():
|
|
self.operators = []
|
|
for i in range(subGraph.OperatorsLength()):
|
|
if subGraph.Operators(i) is None:
|
|
self.operators.append(None)
|
|
else:
|
|
operator_ = OperatorT.InitFromObj(subGraph.Operators(i))
|
|
self.operators.append(operator_)
|
|
self.name = subGraph.Name()
|
|
|
|
# SubGraphT
|
|
def Pack(self, builder):
|
|
if self.tensors is not None:
|
|
tensorslist = []
|
|
for i in range(len(self.tensors)):
|
|
tensorslist.append(self.tensors[i].Pack(builder))
|
|
SubGraphStartTensorsVector(builder, len(self.tensors))
|
|
for i in reversed(range(len(self.tensors))):
|
|
builder.PrependUOffsetTRelative(tensorslist[i])
|
|
tensors = builder.EndVector()
|
|
if self.inputs is not None:
|
|
if np is not None and type(self.inputs) is np.ndarray:
|
|
inputs = builder.CreateNumpyVector(self.inputs)
|
|
else:
|
|
SubGraphStartInputsVector(builder, len(self.inputs))
|
|
for i in reversed(range(len(self.inputs))):
|
|
builder.PrependInt32(self.inputs[i])
|
|
inputs = builder.EndVector()
|
|
if self.outputs is not None:
|
|
if np is not None and type(self.outputs) is np.ndarray:
|
|
outputs = builder.CreateNumpyVector(self.outputs)
|
|
else:
|
|
SubGraphStartOutputsVector(builder, len(self.outputs))
|
|
for i in reversed(range(len(self.outputs))):
|
|
builder.PrependInt32(self.outputs[i])
|
|
outputs = builder.EndVector()
|
|
if self.operators is not None:
|
|
operatorslist = []
|
|
for i in range(len(self.operators)):
|
|
operatorslist.append(self.operators[i].Pack(builder))
|
|
SubGraphStartOperatorsVector(builder, len(self.operators))
|
|
for i in reversed(range(len(self.operators))):
|
|
builder.PrependUOffsetTRelative(operatorslist[i])
|
|
operators = builder.EndVector()
|
|
if self.name is not None:
|
|
name = builder.CreateString(self.name)
|
|
SubGraphStart(builder)
|
|
if self.tensors is not None:
|
|
SubGraphAddTensors(builder, tensors)
|
|
if self.inputs is not None:
|
|
SubGraphAddInputs(builder, inputs)
|
|
if self.outputs is not None:
|
|
SubGraphAddOutputs(builder, outputs)
|
|
if self.operators is not None:
|
|
SubGraphAddOperators(builder, operators)
|
|
if self.name is not None:
|
|
SubGraphAddName(builder, name)
|
|
subGraph = SubGraphEnd(builder)
|
|
return subGraph
|
|
|
|
|
|
class Buffer(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Buffer()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsBuffer(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def BufferBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Buffer
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Buffer
|
|
def Data(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.Uint8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
return 0
|
|
|
|
# Buffer
|
|
def DataAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Uint8Flags, o)
|
|
return 0
|
|
|
|
# Buffer
|
|
def DataLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Buffer
|
|
def DataIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# Buffer
|
|
def Offset(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Buffer
|
|
def Size(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def BufferStart(builder):
|
|
builder.StartObject(3)
|
|
|
|
def BufferAddData(builder, data):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(data), 0)
|
|
|
|
def BufferStartDataVector(builder, numElems):
|
|
return builder.StartVector(1, numElems, 1)
|
|
|
|
def BufferAddOffset(builder, offset):
|
|
builder.PrependUint64Slot(1, offset, 0)
|
|
|
|
def BufferAddSize(builder, size):
|
|
builder.PrependUint64Slot(2, size, 0)
|
|
|
|
def BufferEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class BufferT(object):
|
|
|
|
# BufferT
|
|
def __init__(self):
|
|
self.data = None # type: List[int]
|
|
self.offset = 0 # type: int
|
|
self.size = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
buffer = Buffer()
|
|
buffer.Init(buf, pos)
|
|
return cls.InitFromObj(buffer)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, buffer):
|
|
x = BufferT()
|
|
x._UnPack(buffer)
|
|
return x
|
|
|
|
# BufferT
|
|
def _UnPack(self, buffer):
|
|
if buffer is None:
|
|
return
|
|
if not buffer.DataIsNone():
|
|
if np is None:
|
|
self.data = []
|
|
for i in range(buffer.DataLength()):
|
|
self.data.append(buffer.Data(i))
|
|
else:
|
|
self.data = buffer.DataAsNumpy()
|
|
self.offset = buffer.Offset()
|
|
self.size = buffer.Size()
|
|
|
|
# BufferT
|
|
def Pack(self, builder):
|
|
if self.data is not None:
|
|
if np is not None and type(self.data) is np.ndarray:
|
|
data = builder.CreateNumpyVector(self.data)
|
|
else:
|
|
BufferStartDataVector(builder, len(self.data))
|
|
for i in reversed(range(len(self.data))):
|
|
builder.PrependUint8(self.data[i])
|
|
data = builder.EndVector()
|
|
BufferStart(builder)
|
|
if self.data is not None:
|
|
BufferAddData(builder, data)
|
|
BufferAddOffset(builder, self.offset)
|
|
BufferAddSize(builder, self.size)
|
|
buffer = BufferEnd(builder)
|
|
return buffer
|
|
|
|
|
|
class Metadata(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Metadata()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsMetadata(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def MetadataBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Metadata
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Metadata
|
|
def Name(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# Metadata
|
|
def Buffer(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def MetadataStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def MetadataAddName(builder, name):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
|
|
|
|
def MetadataAddBuffer(builder, buffer):
|
|
builder.PrependUint32Slot(1, buffer, 0)
|
|
|
|
def MetadataEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class MetadataT(object):
|
|
|
|
# MetadataT
|
|
def __init__(self):
|
|
self.name = None # type: str
|
|
self.buffer = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
metadata = Metadata()
|
|
metadata.Init(buf, pos)
|
|
return cls.InitFromObj(metadata)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, metadata):
|
|
x = MetadataT()
|
|
x._UnPack(metadata)
|
|
return x
|
|
|
|
# MetadataT
|
|
def _UnPack(self, metadata):
|
|
if metadata is None:
|
|
return
|
|
self.name = metadata.Name()
|
|
self.buffer = metadata.Buffer()
|
|
|
|
# MetadataT
|
|
def Pack(self, builder):
|
|
if self.name is not None:
|
|
name = builder.CreateString(self.name)
|
|
MetadataStart(builder)
|
|
if self.name is not None:
|
|
MetadataAddName(builder, name)
|
|
MetadataAddBuffer(builder, self.buffer)
|
|
metadata = MetadataEnd(builder)
|
|
return metadata
|
|
|
|
|
|
class TensorMap(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = TensorMap()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsTensorMap(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def TensorMapBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# TensorMap
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# TensorMap
|
|
def Name(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# TensorMap
|
|
def TensorIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def TensorMapStart(builder):
|
|
builder.StartObject(2)
|
|
|
|
def TensorMapAddName(builder, name):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
|
|
|
|
def TensorMapAddTensorIndex(builder, tensorIndex):
|
|
builder.PrependUint32Slot(1, tensorIndex, 0)
|
|
|
|
def TensorMapEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
|
|
class TensorMapT(object):
|
|
|
|
# TensorMapT
|
|
def __init__(self):
|
|
self.name = None # type: str
|
|
self.tensorIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
tensorMap = TensorMap()
|
|
tensorMap.Init(buf, pos)
|
|
return cls.InitFromObj(tensorMap)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, tensorMap):
|
|
x = TensorMapT()
|
|
x._UnPack(tensorMap)
|
|
return x
|
|
|
|
# TensorMapT
|
|
def _UnPack(self, tensorMap):
|
|
if tensorMap is None:
|
|
return
|
|
self.name = tensorMap.Name()
|
|
self.tensorIndex = tensorMap.TensorIndex()
|
|
|
|
# TensorMapT
|
|
def Pack(self, builder):
|
|
if self.name is not None:
|
|
name = builder.CreateString(self.name)
|
|
TensorMapStart(builder)
|
|
if self.name is not None:
|
|
TensorMapAddName(builder, name)
|
|
TensorMapAddTensorIndex(builder, self.tensorIndex)
|
|
tensorMap = TensorMapEnd(builder)
|
|
return tensorMap
|
|
|
|
|
|
class SignatureDef(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = SignatureDef()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsSignatureDef(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def SignatureDefBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# SignatureDef
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# SignatureDef
|
|
def Inputs(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = TensorMap()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# SignatureDef
|
|
def InputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SignatureDef
|
|
def InputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
return o == 0
|
|
|
|
# SignatureDef
|
|
def Outputs(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = TensorMap()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# SignatureDef
|
|
def OutputsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# SignatureDef
|
|
def OutputsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# SignatureDef
|
|
def SignatureKey(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# SignatureDef
|
|
def SubgraphIndex(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
def SignatureDefStart(builder):
|
|
builder.StartObject(5)
|
|
|
|
def SignatureDefAddInputs(builder, inputs):
|
|
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(inputs), 0)
|
|
|
|
def SignatureDefStartInputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SignatureDefAddOutputs(builder, outputs):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(outputs), 0)
|
|
|
|
def SignatureDefStartOutputsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def SignatureDefAddSignatureKey(builder, signatureKey):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(signatureKey), 0)
|
|
|
|
def SignatureDefAddSubgraphIndex(builder, subgraphIndex):
|
|
builder.PrependUint32Slot(4, subgraphIndex, 0)
|
|
|
|
def SignatureDefEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class SignatureDefT(object):
|
|
|
|
# SignatureDefT
|
|
def __init__(self):
|
|
self.inputs = None # type: List[TensorMapT]
|
|
self.outputs = None # type: List[TensorMapT]
|
|
self.signatureKey = None # type: str
|
|
self.subgraphIndex = 0 # type: int
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
signatureDef = SignatureDef()
|
|
signatureDef.Init(buf, pos)
|
|
return cls.InitFromObj(signatureDef)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, signatureDef):
|
|
x = SignatureDefT()
|
|
x._UnPack(signatureDef)
|
|
return x
|
|
|
|
# SignatureDefT
|
|
def _UnPack(self, signatureDef):
|
|
if signatureDef is None:
|
|
return
|
|
if not signatureDef.InputsIsNone():
|
|
self.inputs = []
|
|
for i in range(signatureDef.InputsLength()):
|
|
if signatureDef.Inputs(i) is None:
|
|
self.inputs.append(None)
|
|
else:
|
|
tensorMap_ = TensorMapT.InitFromObj(signatureDef.Inputs(i))
|
|
self.inputs.append(tensorMap_)
|
|
if not signatureDef.OutputsIsNone():
|
|
self.outputs = []
|
|
for i in range(signatureDef.OutputsLength()):
|
|
if signatureDef.Outputs(i) is None:
|
|
self.outputs.append(None)
|
|
else:
|
|
tensorMap_ = TensorMapT.InitFromObj(signatureDef.Outputs(i))
|
|
self.outputs.append(tensorMap_)
|
|
self.signatureKey = signatureDef.SignatureKey()
|
|
self.subgraphIndex = signatureDef.SubgraphIndex()
|
|
|
|
# SignatureDefT
|
|
def Pack(self, builder):
|
|
if self.inputs is not None:
|
|
inputslist = []
|
|
for i in range(len(self.inputs)):
|
|
inputslist.append(self.inputs[i].Pack(builder))
|
|
SignatureDefStartInputsVector(builder, len(self.inputs))
|
|
for i in reversed(range(len(self.inputs))):
|
|
builder.PrependUOffsetTRelative(inputslist[i])
|
|
inputs = builder.EndVector()
|
|
if self.outputs is not None:
|
|
outputslist = []
|
|
for i in range(len(self.outputs)):
|
|
outputslist.append(self.outputs[i].Pack(builder))
|
|
SignatureDefStartOutputsVector(builder, len(self.outputs))
|
|
for i in reversed(range(len(self.outputs))):
|
|
builder.PrependUOffsetTRelative(outputslist[i])
|
|
outputs = builder.EndVector()
|
|
if self.signatureKey is not None:
|
|
signatureKey = builder.CreateString(self.signatureKey)
|
|
SignatureDefStart(builder)
|
|
if self.inputs is not None:
|
|
SignatureDefAddInputs(builder, inputs)
|
|
if self.outputs is not None:
|
|
SignatureDefAddOutputs(builder, outputs)
|
|
if self.signatureKey is not None:
|
|
SignatureDefAddSignatureKey(builder, signatureKey)
|
|
SignatureDefAddSubgraphIndex(builder, self.subgraphIndex)
|
|
signatureDef = SignatureDefEnd(builder)
|
|
return signatureDef
|
|
|
|
|
|
class Model(object):
|
|
__slots__ = ['_tab']
|
|
|
|
@classmethod
|
|
def GetRootAs(cls, buf, offset=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
x = Model()
|
|
x.Init(buf, n + offset)
|
|
return x
|
|
|
|
@classmethod
|
|
def GetRootAsModel(cls, buf, offset=0):
|
|
"""This method is deprecated. Please switch to GetRootAs."""
|
|
return cls.GetRootAs(buf, offset)
|
|
@classmethod
|
|
def ModelBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
|
|
|
|
# Model
|
|
def Init(self, buf, pos):
|
|
self._tab = flatbuffers.table.Table(buf, pos)
|
|
|
|
# Model
|
|
def Version(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
if o != 0:
|
|
return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
|
|
return 0
|
|
|
|
# Model
|
|
def OperatorCodes(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = OperatorCode()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Model
|
|
def OperatorCodesLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def OperatorCodesIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
return o == 0
|
|
|
|
# Model
|
|
def Subgraphs(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = SubGraph()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Model
|
|
def SubgraphsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def SubgraphsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
return o == 0
|
|
|
|
# Model
|
|
def Description(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
if o != 0:
|
|
return self._tab.String(o + self._tab.Pos)
|
|
return None
|
|
|
|
# Model
|
|
def Buffers(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = Buffer()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Model
|
|
def BuffersLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def BuffersIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
return o == 0
|
|
|
|
# Model
|
|
def MetadataBuffer(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
a = self._tab.Vector(o)
|
|
return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
return 0
|
|
|
|
# Model
|
|
def MetadataBufferAsNumpy(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
return 0
|
|
|
|
# Model
|
|
def MetadataBufferLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def MetadataBufferIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
return o == 0
|
|
|
|
# Model
|
|
def Metadata(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = Metadata()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Model
|
|
def MetadataLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def MetadataIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
return o == 0
|
|
|
|
# Model
|
|
def SignatureDefs(self, j):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
x = self._tab.Vector(o)
|
|
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
x = self._tab.Indirect(x)
|
|
obj = SignatureDef()
|
|
obj.Init(self._tab.Bytes, x)
|
|
return obj
|
|
return None
|
|
|
|
# Model
|
|
def SignatureDefsLength(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
if o != 0:
|
|
return self._tab.VectorLen(o)
|
|
return 0
|
|
|
|
# Model
|
|
def SignatureDefsIsNone(self):
|
|
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
return o == 0
|
|
|
|
def ModelStart(builder):
|
|
builder.StartObject(8)
|
|
|
|
def ModelAddVersion(builder, version):
|
|
builder.PrependUint32Slot(0, version, 0)
|
|
|
|
def ModelAddOperatorCodes(builder, operatorCodes):
|
|
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(operatorCodes), 0)
|
|
|
|
def ModelStartOperatorCodesVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelAddSubgraphs(builder, subgraphs):
|
|
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(subgraphs), 0)
|
|
|
|
def ModelStartSubgraphsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelAddDescription(builder, description):
|
|
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(description), 0)
|
|
|
|
def ModelAddBuffers(builder, buffers):
|
|
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(buffers), 0)
|
|
|
|
def ModelStartBuffersVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelAddMetadataBuffer(builder, metadataBuffer):
|
|
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(metadataBuffer), 0)
|
|
|
|
def ModelStartMetadataBufferVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelAddMetadata(builder, metadata):
|
|
builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(metadata), 0)
|
|
|
|
def ModelStartMetadataVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelAddSignatureDefs(builder, signatureDefs):
|
|
builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(signatureDefs), 0)
|
|
|
|
def ModelStartSignatureDefsVector(builder, numElems):
|
|
return builder.StartVector(4, numElems, 4)
|
|
|
|
def ModelEnd(builder):
|
|
return builder.EndObject()
|
|
|
|
|
|
try:
|
|
from typing import List
|
|
except:
|
|
pass
|
|
|
|
class ModelT(object):
|
|
|
|
# ModelT
|
|
def __init__(self):
|
|
self.version = 0 # type: int
|
|
self.operatorCodes = None # type: List[OperatorCodeT]
|
|
self.subgraphs = None # type: List[SubGraphT]
|
|
self.description = None # type: str
|
|
self.buffers = None # type: List[BufferT]
|
|
self.metadataBuffer = None # type: List[int]
|
|
self.metadata = None # type: List[MetadataT]
|
|
self.signatureDefs = None # type: List[SignatureDefT]
|
|
|
|
@classmethod
|
|
def InitFromBuf(cls, buf, pos):
|
|
model = Model()
|
|
model.Init(buf, pos)
|
|
return cls.InitFromObj(model)
|
|
|
|
@classmethod
|
|
def InitFromPackedBuf(cls, buf, pos=0):
|
|
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
return cls.InitFromBuf(buf, pos+n)
|
|
|
|
@classmethod
|
|
def InitFromObj(cls, model):
|
|
x = ModelT()
|
|
x._UnPack(model)
|
|
return x
|
|
|
|
# ModelT
|
|
def _UnPack(self, model):
|
|
if model is None:
|
|
return
|
|
self.version = model.Version()
|
|
if not model.OperatorCodesIsNone():
|
|
self.operatorCodes = []
|
|
for i in range(model.OperatorCodesLength()):
|
|
if model.OperatorCodes(i) is None:
|
|
self.operatorCodes.append(None)
|
|
else:
|
|
operatorCode_ = OperatorCodeT.InitFromObj(model.OperatorCodes(i))
|
|
self.operatorCodes.append(operatorCode_)
|
|
if not model.SubgraphsIsNone():
|
|
self.subgraphs = []
|
|
for i in range(model.SubgraphsLength()):
|
|
if model.Subgraphs(i) is None:
|
|
self.subgraphs.append(None)
|
|
else:
|
|
subGraph_ = SubGraphT.InitFromObj(model.Subgraphs(i))
|
|
self.subgraphs.append(subGraph_)
|
|
self.description = model.Description()
|
|
if not model.BuffersIsNone():
|
|
self.buffers = []
|
|
for i in range(model.BuffersLength()):
|
|
if model.Buffers(i) is None:
|
|
self.buffers.append(None)
|
|
else:
|
|
buffer_ = BufferT.InitFromObj(model.Buffers(i))
|
|
self.buffers.append(buffer_)
|
|
if not model.MetadataBufferIsNone():
|
|
if np is None:
|
|
self.metadataBuffer = []
|
|
for i in range(model.MetadataBufferLength()):
|
|
self.metadataBuffer.append(model.MetadataBuffer(i))
|
|
else:
|
|
self.metadataBuffer = model.MetadataBufferAsNumpy()
|
|
if not model.MetadataIsNone():
|
|
self.metadata = []
|
|
for i in range(model.MetadataLength()):
|
|
if model.Metadata(i) is None:
|
|
self.metadata.append(None)
|
|
else:
|
|
metadata_ = MetadataT.InitFromObj(model.Metadata(i))
|
|
self.metadata.append(metadata_)
|
|
if not model.SignatureDefsIsNone():
|
|
self.signatureDefs = []
|
|
for i in range(model.SignatureDefsLength()):
|
|
if model.SignatureDefs(i) is None:
|
|
self.signatureDefs.append(None)
|
|
else:
|
|
signatureDef_ = SignatureDefT.InitFromObj(model.SignatureDefs(i))
|
|
self.signatureDefs.append(signatureDef_)
|
|
|
|
# ModelT
|
|
def Pack(self, builder):
|
|
if self.operatorCodes is not None:
|
|
operatorCodeslist = []
|
|
for i in range(len(self.operatorCodes)):
|
|
operatorCodeslist.append(self.operatorCodes[i].Pack(builder))
|
|
ModelStartOperatorCodesVector(builder, len(self.operatorCodes))
|
|
for i in reversed(range(len(self.operatorCodes))):
|
|
builder.PrependUOffsetTRelative(operatorCodeslist[i])
|
|
operatorCodes = builder.EndVector()
|
|
if self.subgraphs is not None:
|
|
subgraphslist = []
|
|
for i in range(len(self.subgraphs)):
|
|
subgraphslist.append(self.subgraphs[i].Pack(builder))
|
|
ModelStartSubgraphsVector(builder, len(self.subgraphs))
|
|
for i in reversed(range(len(self.subgraphs))):
|
|
builder.PrependUOffsetTRelative(subgraphslist[i])
|
|
subgraphs = builder.EndVector()
|
|
if self.description is not None:
|
|
description = builder.CreateString(self.description)
|
|
if self.buffers is not None:
|
|
bufferslist = []
|
|
for i in range(len(self.buffers)):
|
|
bufferslist.append(self.buffers[i].Pack(builder))
|
|
ModelStartBuffersVector(builder, len(self.buffers))
|
|
for i in reversed(range(len(self.buffers))):
|
|
builder.PrependUOffsetTRelative(bufferslist[i])
|
|
buffers = builder.EndVector()
|
|
if self.metadataBuffer is not None:
|
|
if np is not None and type(self.metadataBuffer) is np.ndarray:
|
|
metadataBuffer = builder.CreateNumpyVector(self.metadataBuffer)
|
|
else:
|
|
ModelStartMetadataBufferVector(builder, len(self.metadataBuffer))
|
|
for i in reversed(range(len(self.metadataBuffer))):
|
|
builder.PrependInt32(self.metadataBuffer[i])
|
|
metadataBuffer = builder.EndVector()
|
|
if self.metadata is not None:
|
|
metadatalist = []
|
|
for i in range(len(self.metadata)):
|
|
metadatalist.append(self.metadata[i].Pack(builder))
|
|
ModelStartMetadataVector(builder, len(self.metadata))
|
|
for i in reversed(range(len(self.metadata))):
|
|
builder.PrependUOffsetTRelative(metadatalist[i])
|
|
metadata = builder.EndVector()
|
|
if self.signatureDefs is not None:
|
|
signatureDefslist = []
|
|
for i in range(len(self.signatureDefs)):
|
|
signatureDefslist.append(self.signatureDefs[i].Pack(builder))
|
|
ModelStartSignatureDefsVector(builder, len(self.signatureDefs))
|
|
for i in reversed(range(len(self.signatureDefs))):
|
|
builder.PrependUOffsetTRelative(signatureDefslist[i])
|
|
signatureDefs = builder.EndVector()
|
|
ModelStart(builder)
|
|
ModelAddVersion(builder, self.version)
|
|
if self.operatorCodes is not None:
|
|
ModelAddOperatorCodes(builder, operatorCodes)
|
|
if self.subgraphs is not None:
|
|
ModelAddSubgraphs(builder, subgraphs)
|
|
if self.description is not None:
|
|
ModelAddDescription(builder, description)
|
|
if self.buffers is not None:
|
|
ModelAddBuffers(builder, buffers)
|
|
if self.metadataBuffer is not None:
|
|
ModelAddMetadataBuffer(builder, metadataBuffer)
|
|
if self.metadata is not None:
|
|
ModelAddMetadata(builder, metadata)
|
|
if self.signatureDefs is not None:
|
|
ModelAddSignatureDefs(builder, signatureDefs)
|
|
model = ModelEnd(builder)
|
|
return model
|
|
|
|
|