ai-edge-litert-nightly 2.0.4.dev20250929__cp312-cp312-manylinux_2_27_aarch64.whl → 2.0.4.dev20251001__cp312-cp312-manylinux_2_27_aarch64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of ai-edge-litert-nightly might be problematic. Click here for more details.
- ai_edge_litert/__init__.py +1 -1
- ai_edge_litert/libLiteRtRuntimeCApi.so +0 -0
- ai_edge_litert/libpywrap_litert_common.so +0 -0
- ai_edge_litert/schema_py_generated.py +1105 -469
- {ai_edge_litert_nightly-2.0.4.dev20250929.dist-info → ai_edge_litert_nightly-2.0.4.dev20251001.dist-info}/METADATA +1 -1
- {ai_edge_litert_nightly-2.0.4.dev20250929.dist-info → ai_edge_litert_nightly-2.0.4.dev20251001.dist-info}/RECORD +8 -8
- {ai_edge_litert_nightly-2.0.4.dev20250929.dist-info → ai_edge_litert_nightly-2.0.4.dev20251001.dist-info}/WHEEL +0 -0
- {ai_edge_litert_nightly-2.0.4.dev20250929.dist-info → ai_edge_litert_nightly-2.0.4.dev20251001.dist-info}/top_level.txt +0 -0
|
@@ -903,8 +903,11 @@ except:
|
|
|
903
903
|
class CustomQuantizationT(object):
|
|
904
904
|
|
|
905
905
|
# CustomQuantizationT
|
|
906
|
-
def __init__(
|
|
907
|
-
self
|
|
906
|
+
def __init__(
|
|
907
|
+
self,
|
|
908
|
+
custom = None,
|
|
909
|
+
):
|
|
910
|
+
self.custom = custom # type: Optional[List[int]]
|
|
908
911
|
|
|
909
912
|
@classmethod
|
|
910
913
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1015,10 +1018,15 @@ def BlockwiseQuantizationEnd(builder):
|
|
|
1015
1018
|
class BlockwiseQuantizationT(object):
|
|
1016
1019
|
|
|
1017
1020
|
# BlockwiseQuantizationT
|
|
1018
|
-
def __init__(
|
|
1019
|
-
self
|
|
1020
|
-
|
|
1021
|
-
|
|
1021
|
+
def __init__(
|
|
1022
|
+
self,
|
|
1023
|
+
scales = 0,
|
|
1024
|
+
zeroPoints = 0,
|
|
1025
|
+
blockSize = 0,
|
|
1026
|
+
):
|
|
1027
|
+
self.scales = scales # type: int
|
|
1028
|
+
self.zeroPoints = zeroPoints # type: int
|
|
1029
|
+
self.blockSize = blockSize # type: int
|
|
1022
1030
|
|
|
1023
1031
|
@classmethod
|
|
1024
1032
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1257,14 +1265,23 @@ except:
|
|
|
1257
1265
|
class QuantizationParametersT(object):
|
|
1258
1266
|
|
|
1259
1267
|
# QuantizationParametersT
|
|
1260
|
-
def __init__(
|
|
1261
|
-
self
|
|
1262
|
-
|
|
1263
|
-
|
|
1264
|
-
|
|
1265
|
-
|
|
1266
|
-
|
|
1267
|
-
|
|
1268
|
+
def __init__(
|
|
1269
|
+
self,
|
|
1270
|
+
min = None,
|
|
1271
|
+
max = None,
|
|
1272
|
+
scale = None,
|
|
1273
|
+
zeroPoint = None,
|
|
1274
|
+
detailsType = 0,
|
|
1275
|
+
details = None,
|
|
1276
|
+
quantizedDimension = 0,
|
|
1277
|
+
):
|
|
1278
|
+
self.min = min # type: Optional[List[float]]
|
|
1279
|
+
self.max = max # type: Optional[List[float]]
|
|
1280
|
+
self.scale = scale # type: Optional[List[float]]
|
|
1281
|
+
self.zeroPoint = zeroPoint # type: Optional[List[int]]
|
|
1282
|
+
self.detailsType = detailsType # type: int
|
|
1283
|
+
self.details = details # type: Union[None, 'CustomQuantizationT', 'BlockwiseQuantizationT']
|
|
1284
|
+
self.quantizedDimension = quantizedDimension # type: int
|
|
1268
1285
|
|
|
1269
1286
|
@classmethod
|
|
1270
1287
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1442,8 +1459,11 @@ except:
|
|
|
1442
1459
|
class Int32VectorT(object):
|
|
1443
1460
|
|
|
1444
1461
|
# Int32VectorT
|
|
1445
|
-
def __init__(
|
|
1446
|
-
self
|
|
1462
|
+
def __init__(
|
|
1463
|
+
self,
|
|
1464
|
+
values = None,
|
|
1465
|
+
):
|
|
1466
|
+
self.values = values # type: Optional[List[int]]
|
|
1447
1467
|
|
|
1448
1468
|
@classmethod
|
|
1449
1469
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1561,8 +1581,11 @@ except:
|
|
|
1561
1581
|
class Uint16VectorT(object):
|
|
1562
1582
|
|
|
1563
1583
|
# Uint16VectorT
|
|
1564
|
-
def __init__(
|
|
1565
|
-
self
|
|
1584
|
+
def __init__(
|
|
1585
|
+
self,
|
|
1586
|
+
values = None,
|
|
1587
|
+
):
|
|
1588
|
+
self.values = values # type: Optional[List[int]]
|
|
1566
1589
|
|
|
1567
1590
|
@classmethod
|
|
1568
1591
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1680,8 +1703,11 @@ except:
|
|
|
1680
1703
|
class Uint8VectorT(object):
|
|
1681
1704
|
|
|
1682
1705
|
# Uint8VectorT
|
|
1683
|
-
def __init__(
|
|
1684
|
-
self
|
|
1706
|
+
def __init__(
|
|
1707
|
+
self,
|
|
1708
|
+
values = None,
|
|
1709
|
+
):
|
|
1710
|
+
self.values = values # type: Optional[List[int]]
|
|
1685
1711
|
|
|
1686
1712
|
@classmethod
|
|
1687
1713
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -1832,13 +1858,21 @@ except:
|
|
|
1832
1858
|
class DimensionMetadataT(object):
|
|
1833
1859
|
|
|
1834
1860
|
# DimensionMetadataT
|
|
1835
|
-
def __init__(
|
|
1836
|
-
self
|
|
1837
|
-
|
|
1838
|
-
|
|
1839
|
-
|
|
1840
|
-
|
|
1841
|
-
|
|
1861
|
+
def __init__(
|
|
1862
|
+
self,
|
|
1863
|
+
format = 0,
|
|
1864
|
+
denseSize = 0,
|
|
1865
|
+
arraySegmentsType = 0,
|
|
1866
|
+
arraySegments = None,
|
|
1867
|
+
arrayIndicesType = 0,
|
|
1868
|
+
arrayIndices = None,
|
|
1869
|
+
):
|
|
1870
|
+
self.format = format # type: int
|
|
1871
|
+
self.denseSize = denseSize # type: int
|
|
1872
|
+
self.arraySegmentsType = arraySegmentsType # type: int
|
|
1873
|
+
self.arraySegments = arraySegments # type: Union[None, 'Int32VectorT', 'Uint16VectorT', 'Uint8VectorT']
|
|
1874
|
+
self.arrayIndicesType = arrayIndicesType # type: int
|
|
1875
|
+
self.arrayIndices = arrayIndices # type: Union[None, 'Int32VectorT', 'Uint16VectorT', 'Uint8VectorT']
|
|
1842
1876
|
|
|
1843
1877
|
@classmethod
|
|
1844
1878
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -2020,10 +2054,15 @@ except:
|
|
|
2020
2054
|
class SparsityParametersT(object):
|
|
2021
2055
|
|
|
2022
2056
|
# SparsityParametersT
|
|
2023
|
-
def __init__(
|
|
2024
|
-
self
|
|
2025
|
-
|
|
2026
|
-
|
|
2057
|
+
def __init__(
|
|
2058
|
+
self,
|
|
2059
|
+
traversalOrder = None,
|
|
2060
|
+
blockMap = None,
|
|
2061
|
+
dimMetadata = None,
|
|
2062
|
+
):
|
|
2063
|
+
self.traversalOrder = traversalOrder # type: Optional[List[int]]
|
|
2064
|
+
self.blockMap = blockMap # type: Optional[List[int]]
|
|
2065
|
+
self.dimMetadata = dimMetadata # type: Optional[List[DimensionMetadataT]]
|
|
2027
2066
|
|
|
2028
2067
|
@classmethod
|
|
2029
2068
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -2196,10 +2235,15 @@ except:
|
|
|
2196
2235
|
class VariantSubTypeT(object):
|
|
2197
2236
|
|
|
2198
2237
|
# VariantSubTypeT
|
|
2199
|
-
def __init__(
|
|
2200
|
-
self
|
|
2201
|
-
|
|
2202
|
-
|
|
2238
|
+
def __init__(
|
|
2239
|
+
self,
|
|
2240
|
+
shape = None,
|
|
2241
|
+
type = 0,
|
|
2242
|
+
hasRank = False,
|
|
2243
|
+
):
|
|
2244
|
+
self.shape = shape # type: Optional[List[int]]
|
|
2245
|
+
self.type = type # type: int
|
|
2246
|
+
self.hasRank = hasRank # type: bool
|
|
2203
2247
|
|
|
2204
2248
|
@classmethod
|
|
2205
2249
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -2460,17 +2504,29 @@ except:
|
|
|
2460
2504
|
class TensorT(object):
|
|
2461
2505
|
|
|
2462
2506
|
# TensorT
|
|
2463
|
-
def __init__(
|
|
2464
|
-
self
|
|
2465
|
-
|
|
2466
|
-
|
|
2467
|
-
|
|
2468
|
-
|
|
2469
|
-
|
|
2470
|
-
|
|
2471
|
-
|
|
2472
|
-
|
|
2473
|
-
|
|
2507
|
+
def __init__(
|
|
2508
|
+
self,
|
|
2509
|
+
shape = None,
|
|
2510
|
+
type = 0,
|
|
2511
|
+
buffer = 0,
|
|
2512
|
+
name = None,
|
|
2513
|
+
quantization = None,
|
|
2514
|
+
isVariable = False,
|
|
2515
|
+
sparsity = None,
|
|
2516
|
+
shapeSignature = None,
|
|
2517
|
+
hasRank = False,
|
|
2518
|
+
variantTensors = None,
|
|
2519
|
+
):
|
|
2520
|
+
self.shape = shape # type: Optional[List[int]]
|
|
2521
|
+
self.type = type # type: int
|
|
2522
|
+
self.buffer = buffer # type: int
|
|
2523
|
+
self.name = name # type: Optional[str]
|
|
2524
|
+
self.quantization = quantization # type: Optional[QuantizationParametersT]
|
|
2525
|
+
self.isVariable = isVariable # type: bool
|
|
2526
|
+
self.sparsity = sparsity # type: Optional[SparsityParametersT]
|
|
2527
|
+
self.shapeSignature = shapeSignature # type: Optional[List[int]]
|
|
2528
|
+
self.hasRank = hasRank # type: bool
|
|
2529
|
+
self.variantTensors = variantTensors # type: Optional[List[VariantSubTypeT]]
|
|
2474
2530
|
|
|
2475
2531
|
@classmethod
|
|
2476
2532
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -2767,13 +2823,21 @@ except:
|
|
|
2767
2823
|
class StablehloGatherOptionsT(object):
|
|
2768
2824
|
|
|
2769
2825
|
# StablehloGatherOptionsT
|
|
2770
|
-
def __init__(
|
|
2771
|
-
self
|
|
2772
|
-
|
|
2773
|
-
|
|
2774
|
-
|
|
2775
|
-
|
|
2776
|
-
|
|
2826
|
+
def __init__(
|
|
2827
|
+
self,
|
|
2828
|
+
offsetDims = None,
|
|
2829
|
+
collapsedSliceDims = None,
|
|
2830
|
+
startIndexMap = None,
|
|
2831
|
+
indexVectorDim = 0,
|
|
2832
|
+
sliceSizes = None,
|
|
2833
|
+
indicesAreSorted = False,
|
|
2834
|
+
):
|
|
2835
|
+
self.offsetDims = offsetDims # type: Optional[List[int]]
|
|
2836
|
+
self.collapsedSliceDims = collapsedSliceDims # type: Optional[List[int]]
|
|
2837
|
+
self.startIndexMap = startIndexMap # type: Optional[List[int]]
|
|
2838
|
+
self.indexVectorDim = indexVectorDim # type: int
|
|
2839
|
+
self.sliceSizes = sliceSizes # type: Optional[List[int]]
|
|
2840
|
+
self.indicesAreSorted = indicesAreSorted # type: bool
|
|
2777
2841
|
|
|
2778
2842
|
@classmethod
|
|
2779
2843
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -2946,8 +3010,11 @@ except:
|
|
|
2946
3010
|
class StablehloTransposeOptionsT(object):
|
|
2947
3011
|
|
|
2948
3012
|
# StablehloTransposeOptionsT
|
|
2949
|
-
def __init__(
|
|
2950
|
-
self
|
|
3013
|
+
def __init__(
|
|
3014
|
+
self,
|
|
3015
|
+
permutation = None,
|
|
3016
|
+
):
|
|
3017
|
+
self.permutation = permutation # type: Optional[List[int]]
|
|
2951
3018
|
|
|
2952
3019
|
@classmethod
|
|
2953
3020
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3197,12 +3264,19 @@ except:
|
|
|
3197
3264
|
class StablehloDotGeneralOptionsT(object):
|
|
3198
3265
|
|
|
3199
3266
|
# StablehloDotGeneralOptionsT
|
|
3200
|
-
def __init__(
|
|
3201
|
-
self
|
|
3202
|
-
|
|
3203
|
-
|
|
3204
|
-
|
|
3205
|
-
|
|
3267
|
+
def __init__(
|
|
3268
|
+
self,
|
|
3269
|
+
lhsBatchingDimensions = None,
|
|
3270
|
+
rhsBatchingDimensions = None,
|
|
3271
|
+
lhsContractingDimensions = None,
|
|
3272
|
+
rhsContractingDimensions = None,
|
|
3273
|
+
precisionConfig = None,
|
|
3274
|
+
):
|
|
3275
|
+
self.lhsBatchingDimensions = lhsBatchingDimensions # type: Optional[List[int]]
|
|
3276
|
+
self.rhsBatchingDimensions = rhsBatchingDimensions # type: Optional[List[int]]
|
|
3277
|
+
self.lhsContractingDimensions = lhsContractingDimensions # type: Optional[List[int]]
|
|
3278
|
+
self.rhsContractingDimensions = rhsContractingDimensions # type: Optional[List[int]]
|
|
3279
|
+
self.precisionConfig = precisionConfig # type: Optional[List[int]]
|
|
3206
3280
|
|
|
3207
3281
|
@classmethod
|
|
3208
3282
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3530,13 +3604,21 @@ except:
|
|
|
3530
3604
|
class StablehloReduceWindowOptionsT(object):
|
|
3531
3605
|
|
|
3532
3606
|
# StablehloReduceWindowOptionsT
|
|
3533
|
-
def __init__(
|
|
3534
|
-
self
|
|
3535
|
-
|
|
3536
|
-
|
|
3537
|
-
|
|
3538
|
-
|
|
3539
|
-
|
|
3607
|
+
def __init__(
|
|
3608
|
+
self,
|
|
3609
|
+
windowDimensions = None,
|
|
3610
|
+
windowStrides = None,
|
|
3611
|
+
baseDilations = None,
|
|
3612
|
+
windowDilations = None,
|
|
3613
|
+
padding = None,
|
|
3614
|
+
bodySubgraphIndex = 0,
|
|
3615
|
+
):
|
|
3616
|
+
self.windowDimensions = windowDimensions # type: Optional[List[int]]
|
|
3617
|
+
self.windowStrides = windowStrides # type: Optional[List[int]]
|
|
3618
|
+
self.baseDilations = baseDilations # type: Optional[List[int]]
|
|
3619
|
+
self.windowDilations = windowDilations # type: Optional[List[int]]
|
|
3620
|
+
self.padding = padding # type: Optional[List[int]]
|
|
3621
|
+
self.bodySubgraphIndex = bodySubgraphIndex # type: int
|
|
3540
3622
|
|
|
3541
3623
|
@classmethod
|
|
3542
3624
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3707,9 +3789,13 @@ def StablehloWhileOptionsEnd(builder):
|
|
|
3707
3789
|
class StablehloWhileOptionsT(object):
|
|
3708
3790
|
|
|
3709
3791
|
# StablehloWhileOptionsT
|
|
3710
|
-
def __init__(
|
|
3711
|
-
self
|
|
3712
|
-
|
|
3792
|
+
def __init__(
|
|
3793
|
+
self,
|
|
3794
|
+
condSubgraphIndex = 0,
|
|
3795
|
+
bodySubgraphIndex = 0,
|
|
3796
|
+
):
|
|
3797
|
+
self.condSubgraphIndex = condSubgraphIndex # type: int
|
|
3798
|
+
self.bodySubgraphIndex = bodySubgraphIndex # type: int
|
|
3713
3799
|
|
|
3714
3800
|
@classmethod
|
|
3715
3801
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3807,10 +3893,15 @@ def StablehloSortOptionsEnd(builder):
|
|
|
3807
3893
|
class StablehloSortOptionsT(object):
|
|
3808
3894
|
|
|
3809
3895
|
# StablehloSortOptionsT
|
|
3810
|
-
def __init__(
|
|
3811
|
-
self
|
|
3812
|
-
|
|
3813
|
-
|
|
3896
|
+
def __init__(
|
|
3897
|
+
self,
|
|
3898
|
+
dimension = 0,
|
|
3899
|
+
isStable = False,
|
|
3900
|
+
comparatorSubgraphIndex = 0,
|
|
3901
|
+
):
|
|
3902
|
+
self.dimension = dimension # type: int
|
|
3903
|
+
self.isStable = isStable # type: bool
|
|
3904
|
+
self.comparatorSubgraphIndex = comparatorSubgraphIndex # type: int
|
|
3814
3905
|
|
|
3815
3906
|
@classmethod
|
|
3816
3907
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3890,8 +3981,11 @@ def StablehloConcatenateOptionsEnd(builder):
|
|
|
3890
3981
|
class StablehloConcatenateOptionsT(object):
|
|
3891
3982
|
|
|
3892
3983
|
# StablehloConcatenateOptionsT
|
|
3893
|
-
def __init__(
|
|
3894
|
-
self
|
|
3984
|
+
def __init__(
|
|
3985
|
+
self,
|
|
3986
|
+
dimension = 0,
|
|
3987
|
+
):
|
|
3988
|
+
self.dimension = dimension # type: int
|
|
3895
3989
|
|
|
3896
3990
|
@classmethod
|
|
3897
3991
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -3994,8 +4088,11 @@ except:
|
|
|
3994
4088
|
class StablehloBroadcastInDimOptionsT(object):
|
|
3995
4089
|
|
|
3996
4090
|
# StablehloBroadcastInDimOptionsT
|
|
3997
|
-
def __init__(
|
|
3998
|
-
self
|
|
4091
|
+
def __init__(
|
|
4092
|
+
self,
|
|
4093
|
+
broadcastDimensions = None,
|
|
4094
|
+
):
|
|
4095
|
+
self.broadcastDimensions = broadcastDimensions # type: Optional[List[int]]
|
|
3999
4096
|
|
|
4000
4097
|
@classmethod
|
|
4001
4098
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4096,9 +4193,13 @@ def StablehloCompareOptionsEnd(builder):
|
|
|
4096
4193
|
class StablehloCompareOptionsT(object):
|
|
4097
4194
|
|
|
4098
4195
|
# StablehloCompareOptionsT
|
|
4099
|
-
def __init__(
|
|
4100
|
-
self
|
|
4101
|
-
|
|
4196
|
+
def __init__(
|
|
4197
|
+
self,
|
|
4198
|
+
comparisonDirection = 0,
|
|
4199
|
+
compareType = 0,
|
|
4200
|
+
):
|
|
4201
|
+
self.comparisonDirection = comparisonDirection # type: int
|
|
4202
|
+
self.compareType = compareType # type: int
|
|
4102
4203
|
|
|
4103
4204
|
@classmethod
|
|
4104
4205
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4203,8 +4304,11 @@ except:
|
|
|
4203
4304
|
class StablehloDynamicSliceOptionsT(object):
|
|
4204
4305
|
|
|
4205
4306
|
# StablehloDynamicSliceOptionsT
|
|
4206
|
-
def __init__(
|
|
4207
|
-
self
|
|
4307
|
+
def __init__(
|
|
4308
|
+
self,
|
|
4309
|
+
sliceSizes = None,
|
|
4310
|
+
):
|
|
4311
|
+
self.sliceSizes = sliceSizes # type: Optional[List[int]]
|
|
4208
4312
|
|
|
4209
4313
|
@classmethod
|
|
4210
4314
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4388,10 +4492,15 @@ except:
|
|
|
4388
4492
|
class StablehloPadOptionsT(object):
|
|
4389
4493
|
|
|
4390
4494
|
# StablehloPadOptionsT
|
|
4391
|
-
def __init__(
|
|
4392
|
-
self
|
|
4393
|
-
|
|
4394
|
-
|
|
4495
|
+
def __init__(
|
|
4496
|
+
self,
|
|
4497
|
+
edgePaddingLow = None,
|
|
4498
|
+
edgePaddingHigh = None,
|
|
4499
|
+
interiorPadding = None,
|
|
4500
|
+
):
|
|
4501
|
+
self.edgePaddingLow = edgePaddingLow # type: Optional[List[int]]
|
|
4502
|
+
self.edgePaddingHigh = edgePaddingHigh # type: Optional[List[int]]
|
|
4503
|
+
self.interiorPadding = interiorPadding # type: Optional[List[int]]
|
|
4395
4504
|
|
|
4396
4505
|
@classmethod
|
|
4397
4506
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4516,8 +4625,11 @@ def StablehloIotaOptionsEnd(builder):
|
|
|
4516
4625
|
class StablehloIotaOptionsT(object):
|
|
4517
4626
|
|
|
4518
4627
|
# StablehloIotaOptionsT
|
|
4519
|
-
def __init__(
|
|
4520
|
-
self
|
|
4628
|
+
def __init__(
|
|
4629
|
+
self,
|
|
4630
|
+
iotaDimension = 0,
|
|
4631
|
+
):
|
|
4632
|
+
self.iotaDimension = iotaDimension # type: int
|
|
4521
4633
|
|
|
4522
4634
|
@classmethod
|
|
4523
4635
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4693,13 +4805,21 @@ except:
|
|
|
4693
4805
|
class StablehloCustomCallOptionsT(object):
|
|
4694
4806
|
|
|
4695
4807
|
# StablehloCustomCallOptionsT
|
|
4696
|
-
def __init__(
|
|
4697
|
-
self
|
|
4698
|
-
|
|
4699
|
-
|
|
4700
|
-
|
|
4701
|
-
|
|
4702
|
-
|
|
4808
|
+
def __init__(
|
|
4809
|
+
self,
|
|
4810
|
+
callTargetName = None,
|
|
4811
|
+
hasSideEffect = False,
|
|
4812
|
+
backendConfig = None,
|
|
4813
|
+
apiVersion = 0,
|
|
4814
|
+
calledComputations = None,
|
|
4815
|
+
customAttributes = None,
|
|
4816
|
+
):
|
|
4817
|
+
self.callTargetName = callTargetName # type: Optional[str]
|
|
4818
|
+
self.hasSideEffect = hasSideEffect # type: bool
|
|
4819
|
+
self.backendConfig = backendConfig # type: Optional[str]
|
|
4820
|
+
self.apiVersion = apiVersion # type: int
|
|
4821
|
+
self.calledComputations = calledComputations # type: Optional[List[int]]
|
|
4822
|
+
self.customAttributes = customAttributes # type: Optional[List[int]]
|
|
4703
4823
|
|
|
4704
4824
|
@classmethod
|
|
4705
4825
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -4858,9 +4978,13 @@ except:
|
|
|
4858
4978
|
class StablehloReduceOptionsT(object):
|
|
4859
4979
|
|
|
4860
4980
|
# StablehloReduceOptionsT
|
|
4861
|
-
def __init__(
|
|
4862
|
-
self
|
|
4863
|
-
|
|
4981
|
+
def __init__(
|
|
4982
|
+
self,
|
|
4983
|
+
dimensions = None,
|
|
4984
|
+
bodySubgraphIndex = 0,
|
|
4985
|
+
):
|
|
4986
|
+
self.dimensions = dimensions # type: Optional[List[int]]
|
|
4987
|
+
self.bodySubgraphIndex = bodySubgraphIndex # type: int
|
|
4864
4988
|
|
|
4865
4989
|
@classmethod
|
|
4866
4990
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -5046,10 +5170,15 @@ except:
|
|
|
5046
5170
|
class StablehloSliceOptionsT(object):
|
|
5047
5171
|
|
|
5048
5172
|
# StablehloSliceOptionsT
|
|
5049
|
-
def __init__(
|
|
5050
|
-
self
|
|
5051
|
-
|
|
5052
|
-
|
|
5173
|
+
def __init__(
|
|
5174
|
+
self,
|
|
5175
|
+
startIndices = None,
|
|
5176
|
+
limitIndices = None,
|
|
5177
|
+
strides = None,
|
|
5178
|
+
):
|
|
5179
|
+
self.startIndices = startIndices # type: Optional[List[int]]
|
|
5180
|
+
self.limitIndices = limitIndices # type: Optional[List[int]]
|
|
5181
|
+
self.strides = strides # type: Optional[List[int]]
|
|
5053
5182
|
|
|
5054
5183
|
@classmethod
|
|
5055
5184
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -5545,24 +5674,43 @@ except:
|
|
|
5545
5674
|
class StablehloConvolutionOptionsT(object):
|
|
5546
5675
|
|
|
5547
5676
|
# StablehloConvolutionOptionsT
|
|
5548
|
-
def __init__(
|
|
5549
|
-
self
|
|
5550
|
-
|
|
5551
|
-
|
|
5552
|
-
|
|
5553
|
-
|
|
5554
|
-
|
|
5555
|
-
|
|
5556
|
-
|
|
5557
|
-
|
|
5558
|
-
|
|
5559
|
-
|
|
5560
|
-
|
|
5561
|
-
|
|
5562
|
-
|
|
5563
|
-
|
|
5564
|
-
|
|
5565
|
-
|
|
5677
|
+
def __init__(
|
|
5678
|
+
self,
|
|
5679
|
+
windowStrides = None,
|
|
5680
|
+
padding = None,
|
|
5681
|
+
lhsDilation = None,
|
|
5682
|
+
rhsDilation = None,
|
|
5683
|
+
windowReversal = None,
|
|
5684
|
+
inputBatchDimension = 0,
|
|
5685
|
+
inputFeatureDimension = 0,
|
|
5686
|
+
inputSpatialDimensions = None,
|
|
5687
|
+
kernelInputFeatureDimension = 0,
|
|
5688
|
+
kernelOutputFeatureDimension = 0,
|
|
5689
|
+
kernelSpatialDimensions = None,
|
|
5690
|
+
outputBatchDimension = 0,
|
|
5691
|
+
outputFeatureDimension = 0,
|
|
5692
|
+
outputSpatialDimensions = None,
|
|
5693
|
+
featureGroupCount = 0,
|
|
5694
|
+
batchGroupCount = 0,
|
|
5695
|
+
precisionConfig = None,
|
|
5696
|
+
):
|
|
5697
|
+
self.windowStrides = windowStrides # type: Optional[List[int]]
|
|
5698
|
+
self.padding = padding # type: Optional[List[int]]
|
|
5699
|
+
self.lhsDilation = lhsDilation # type: Optional[List[int]]
|
|
5700
|
+
self.rhsDilation = rhsDilation # type: Optional[List[int]]
|
|
5701
|
+
self.windowReversal = windowReversal # type: Optional[List[bool]]
|
|
5702
|
+
self.inputBatchDimension = inputBatchDimension # type: int
|
|
5703
|
+
self.inputFeatureDimension = inputFeatureDimension # type: int
|
|
5704
|
+
self.inputSpatialDimensions = inputSpatialDimensions # type: Optional[List[int]]
|
|
5705
|
+
self.kernelInputFeatureDimension = kernelInputFeatureDimension # type: int
|
|
5706
|
+
self.kernelOutputFeatureDimension = kernelOutputFeatureDimension # type: int
|
|
5707
|
+
self.kernelSpatialDimensions = kernelSpatialDimensions # type: Optional[List[int]]
|
|
5708
|
+
self.outputBatchDimension = outputBatchDimension # type: int
|
|
5709
|
+
self.outputFeatureDimension = outputFeatureDimension # type: int
|
|
5710
|
+
self.outputSpatialDimensions = outputSpatialDimensions # type: Optional[List[int]]
|
|
5711
|
+
self.featureGroupCount = featureGroupCount # type: int
|
|
5712
|
+
self.batchGroupCount = batchGroupCount # type: int
|
|
5713
|
+
self.precisionConfig = precisionConfig # type: Optional[List[int]]
|
|
5566
5714
|
|
|
5567
5715
|
@classmethod
|
|
5568
5716
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -5938,14 +6086,23 @@ except:
|
|
|
5938
6086
|
class StablehloScatterOptionsT(object):
|
|
5939
6087
|
|
|
5940
6088
|
# StablehloScatterOptionsT
|
|
5941
|
-
def __init__(
|
|
5942
|
-
self
|
|
5943
|
-
|
|
5944
|
-
|
|
5945
|
-
|
|
5946
|
-
|
|
5947
|
-
|
|
5948
|
-
|
|
6089
|
+
def __init__(
|
|
6090
|
+
self,
|
|
6091
|
+
indicesAreSorted = False,
|
|
6092
|
+
updateWindowDims = None,
|
|
6093
|
+
insertedWindowDims = None,
|
|
6094
|
+
scatterDimsToOperandDims = None,
|
|
6095
|
+
indexVectorDim = 0,
|
|
6096
|
+
uniqueIndices = False,
|
|
6097
|
+
updateComputationSubgraphIndex = 0,
|
|
6098
|
+
):
|
|
6099
|
+
self.indicesAreSorted = indicesAreSorted # type: bool
|
|
6100
|
+
self.updateWindowDims = updateWindowDims # type: Optional[List[int]]
|
|
6101
|
+
self.insertedWindowDims = insertedWindowDims # type: Optional[List[int]]
|
|
6102
|
+
self.scatterDimsToOperandDims = scatterDimsToOperandDims # type: Optional[List[int]]
|
|
6103
|
+
self.indexVectorDim = indexVectorDim # type: int
|
|
6104
|
+
self.uniqueIndices = uniqueIndices # type: bool
|
|
6105
|
+
self.updateComputationSubgraphIndex = updateComputationSubgraphIndex # type: int
|
|
5949
6106
|
|
|
5950
6107
|
@classmethod
|
|
5951
6108
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6105,8 +6262,11 @@ except:
|
|
|
6105
6262
|
class StablehloCaseOptionsT(object):
|
|
6106
6263
|
|
|
6107
6264
|
# StablehloCaseOptionsT
|
|
6108
|
-
def __init__(
|
|
6109
|
-
self
|
|
6265
|
+
def __init__(
|
|
6266
|
+
self,
|
|
6267
|
+
branchSubgraphIndices = None,
|
|
6268
|
+
):
|
|
6269
|
+
self.branchSubgraphIndices = branchSubgraphIndices # type: Optional[List[int]]
|
|
6110
6270
|
|
|
6111
6271
|
@classmethod
|
|
6112
6272
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6197,8 +6357,11 @@ def StablehloRngBitGeneratorOptionsEnd(builder):
|
|
|
6197
6357
|
class StablehloRngBitGeneratorOptionsT(object):
|
|
6198
6358
|
|
|
6199
6359
|
# StablehloRngBitGeneratorOptionsT
|
|
6200
|
-
def __init__(
|
|
6201
|
-
self
|
|
6360
|
+
def __init__(
|
|
6361
|
+
self,
|
|
6362
|
+
algorithm = 0,
|
|
6363
|
+
):
|
|
6364
|
+
self.algorithm = algorithm # type: int
|
|
6202
6365
|
|
|
6203
6366
|
@classmethod
|
|
6204
6367
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6334,14 +6497,23 @@ def Conv2DOptionsEnd(builder):
|
|
|
6334
6497
|
class Conv2DOptionsT(object):
|
|
6335
6498
|
|
|
6336
6499
|
# Conv2DOptionsT
|
|
6337
|
-
def __init__(
|
|
6338
|
-
self
|
|
6339
|
-
|
|
6340
|
-
|
|
6341
|
-
|
|
6342
|
-
|
|
6343
|
-
|
|
6344
|
-
|
|
6500
|
+
def __init__(
|
|
6501
|
+
self,
|
|
6502
|
+
padding = 0,
|
|
6503
|
+
strideW = 0,
|
|
6504
|
+
strideH = 0,
|
|
6505
|
+
fusedActivationFunction = 0,
|
|
6506
|
+
dilationWFactor = 1,
|
|
6507
|
+
dilationHFactor = 1,
|
|
6508
|
+
quantizedBiasType = 0,
|
|
6509
|
+
):
|
|
6510
|
+
self.padding = padding # type: int
|
|
6511
|
+
self.strideW = strideW # type: int
|
|
6512
|
+
self.strideH = strideH # type: int
|
|
6513
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
6514
|
+
self.dilationWFactor = dilationWFactor # type: int
|
|
6515
|
+
self.dilationHFactor = dilationHFactor # type: int
|
|
6516
|
+
self.quantizedBiasType = quantizedBiasType # type: int
|
|
6345
6517
|
|
|
6346
6518
|
@classmethod
|
|
6347
6519
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6499,15 +6671,25 @@ def Conv3DOptionsEnd(builder):
|
|
|
6499
6671
|
class Conv3DOptionsT(object):
|
|
6500
6672
|
|
|
6501
6673
|
# Conv3DOptionsT
|
|
6502
|
-
def __init__(
|
|
6503
|
-
self
|
|
6504
|
-
|
|
6505
|
-
|
|
6506
|
-
|
|
6507
|
-
|
|
6508
|
-
|
|
6509
|
-
|
|
6510
|
-
|
|
6674
|
+
def __init__(
|
|
6675
|
+
self,
|
|
6676
|
+
padding = 0,
|
|
6677
|
+
strideD = 0,
|
|
6678
|
+
strideW = 0,
|
|
6679
|
+
strideH = 0,
|
|
6680
|
+
fusedActivationFunction = 0,
|
|
6681
|
+
dilationDFactor = 1,
|
|
6682
|
+
dilationWFactor = 1,
|
|
6683
|
+
dilationHFactor = 1,
|
|
6684
|
+
):
|
|
6685
|
+
self.padding = padding # type: int
|
|
6686
|
+
self.strideD = strideD # type: int
|
|
6687
|
+
self.strideW = strideW # type: int
|
|
6688
|
+
self.strideH = strideH # type: int
|
|
6689
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
6690
|
+
self.dilationDFactor = dilationDFactor # type: int
|
|
6691
|
+
self.dilationWFactor = dilationWFactor # type: int
|
|
6692
|
+
self.dilationHFactor = dilationHFactor # type: int
|
|
6511
6693
|
|
|
6512
6694
|
@classmethod
|
|
6513
6695
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6647,13 +6829,21 @@ def Pool2DOptionsEnd(builder):
|
|
|
6647
6829
|
class Pool2DOptionsT(object):
|
|
6648
6830
|
|
|
6649
6831
|
# Pool2DOptionsT
|
|
6650
|
-
def __init__(
|
|
6651
|
-
self
|
|
6652
|
-
|
|
6653
|
-
|
|
6654
|
-
|
|
6655
|
-
|
|
6656
|
-
|
|
6832
|
+
def __init__(
|
|
6833
|
+
self,
|
|
6834
|
+
padding = 0,
|
|
6835
|
+
strideW = 0,
|
|
6836
|
+
strideH = 0,
|
|
6837
|
+
filterWidth = 0,
|
|
6838
|
+
filterHeight = 0,
|
|
6839
|
+
fusedActivationFunction = 0,
|
|
6840
|
+
):
|
|
6841
|
+
self.padding = padding # type: int
|
|
6842
|
+
self.strideW = strideW # type: int
|
|
6843
|
+
self.strideH = strideH # type: int
|
|
6844
|
+
self.filterWidth = filterWidth # type: int
|
|
6845
|
+
self.filterHeight = filterHeight # type: int
|
|
6846
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
6657
6847
|
|
|
6658
6848
|
@classmethod
|
|
6659
6849
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6799,14 +6989,23 @@ def DepthwiseConv2DOptionsEnd(builder):
|
|
|
6799
6989
|
class DepthwiseConv2DOptionsT(object):
|
|
6800
6990
|
|
|
6801
6991
|
# DepthwiseConv2DOptionsT
|
|
6802
|
-
def __init__(
|
|
6803
|
-
self
|
|
6804
|
-
|
|
6805
|
-
|
|
6806
|
-
|
|
6807
|
-
|
|
6808
|
-
|
|
6809
|
-
|
|
6992
|
+
def __init__(
|
|
6993
|
+
self,
|
|
6994
|
+
padding = 0,
|
|
6995
|
+
strideW = 0,
|
|
6996
|
+
strideH = 0,
|
|
6997
|
+
depthMultiplier = 0,
|
|
6998
|
+
fusedActivationFunction = 0,
|
|
6999
|
+
dilationWFactor = 1,
|
|
7000
|
+
dilationHFactor = 1,
|
|
7001
|
+
):
|
|
7002
|
+
self.padding = padding # type: int
|
|
7003
|
+
self.strideW = strideW # type: int
|
|
7004
|
+
self.strideH = strideH # type: int
|
|
7005
|
+
self.depthMultiplier = depthMultiplier # type: int
|
|
7006
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7007
|
+
self.dilationWFactor = dilationWFactor # type: int
|
|
7008
|
+
self.dilationHFactor = dilationHFactor # type: int
|
|
6810
7009
|
|
|
6811
7010
|
@classmethod
|
|
6812
7011
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -6964,10 +7163,15 @@ except:
|
|
|
6964
7163
|
class ConcatEmbeddingsOptionsT(object):
|
|
6965
7164
|
|
|
6966
7165
|
# ConcatEmbeddingsOptionsT
|
|
6967
|
-
def __init__(
|
|
6968
|
-
self
|
|
6969
|
-
|
|
6970
|
-
|
|
7166
|
+
def __init__(
|
|
7167
|
+
self,
|
|
7168
|
+
numChannels = 0,
|
|
7169
|
+
numColumnsPerChannel = None,
|
|
7170
|
+
embeddingDimPerChannel = None,
|
|
7171
|
+
):
|
|
7172
|
+
self.numChannels = numChannels # type: int
|
|
7173
|
+
self.numColumnsPerChannel = numColumnsPerChannel # type: Optional[List[int]]
|
|
7174
|
+
self.embeddingDimPerChannel = embeddingDimPerChannel # type: Optional[List[int]]
|
|
6971
7175
|
|
|
6972
7176
|
@classmethod
|
|
6973
7177
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7077,8 +7281,11 @@ def LSHProjectionOptionsEnd(builder):
|
|
|
7077
7281
|
class LSHProjectionOptionsT(object):
|
|
7078
7282
|
|
|
7079
7283
|
# LSHProjectionOptionsT
|
|
7080
|
-
def __init__(
|
|
7081
|
-
self
|
|
7284
|
+
def __init__(
|
|
7285
|
+
self,
|
|
7286
|
+
type = 0,
|
|
7287
|
+
):
|
|
7288
|
+
self.type = type # type: int
|
|
7082
7289
|
|
|
7083
7290
|
@classmethod
|
|
7084
7291
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7174,10 +7381,15 @@ def SVDFOptionsEnd(builder):
|
|
|
7174
7381
|
class SVDFOptionsT(object):
|
|
7175
7382
|
|
|
7176
7383
|
# SVDFOptionsT
|
|
7177
|
-
def __init__(
|
|
7178
|
-
self
|
|
7179
|
-
|
|
7180
|
-
|
|
7384
|
+
def __init__(
|
|
7385
|
+
self,
|
|
7386
|
+
rank = 0,
|
|
7387
|
+
fusedActivationFunction = 0,
|
|
7388
|
+
asymmetricQuantizeInputs = False,
|
|
7389
|
+
):
|
|
7390
|
+
self.rank = rank # type: int
|
|
7391
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7392
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
7181
7393
|
|
|
7182
7394
|
@classmethod
|
|
7183
7395
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7267,9 +7479,13 @@ def RNNOptionsEnd(builder):
|
|
|
7267
7479
|
class RNNOptionsT(object):
|
|
7268
7480
|
|
|
7269
7481
|
# RNNOptionsT
|
|
7270
|
-
def __init__(
|
|
7271
|
-
self
|
|
7272
|
-
|
|
7482
|
+
def __init__(
|
|
7483
|
+
self,
|
|
7484
|
+
fusedActivationFunction = 0,
|
|
7485
|
+
asymmetricQuantizeInputs = False,
|
|
7486
|
+
):
|
|
7487
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7488
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
7273
7489
|
|
|
7274
7490
|
@classmethod
|
|
7275
7491
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7367,10 +7583,15 @@ def SequenceRNNOptionsEnd(builder):
|
|
|
7367
7583
|
class SequenceRNNOptionsT(object):
|
|
7368
7584
|
|
|
7369
7585
|
# SequenceRNNOptionsT
|
|
7370
|
-
def __init__(
|
|
7371
|
-
self
|
|
7372
|
-
|
|
7373
|
-
|
|
7586
|
+
def __init__(
|
|
7587
|
+
self,
|
|
7588
|
+
timeMajor = False,
|
|
7589
|
+
fusedActivationFunction = 0,
|
|
7590
|
+
asymmetricQuantizeInputs = False,
|
|
7591
|
+
):
|
|
7592
|
+
self.timeMajor = timeMajor # type: bool
|
|
7593
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7594
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
7374
7595
|
|
|
7375
7596
|
@classmethod
|
|
7376
7597
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7480,11 +7701,17 @@ def BidirectionalSequenceRNNOptionsEnd(builder):
|
|
|
7480
7701
|
class BidirectionalSequenceRNNOptionsT(object):
|
|
7481
7702
|
|
|
7482
7703
|
# BidirectionalSequenceRNNOptionsT
|
|
7483
|
-
def __init__(
|
|
7484
|
-
self
|
|
7485
|
-
|
|
7486
|
-
|
|
7487
|
-
|
|
7704
|
+
def __init__(
|
|
7705
|
+
self,
|
|
7706
|
+
timeMajor = False,
|
|
7707
|
+
fusedActivationFunction = 0,
|
|
7708
|
+
mergeOutputs = False,
|
|
7709
|
+
asymmetricQuantizeInputs = False,
|
|
7710
|
+
):
|
|
7711
|
+
self.timeMajor = timeMajor # type: bool
|
|
7712
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7713
|
+
self.mergeOutputs = mergeOutputs # type: bool
|
|
7714
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
7488
7715
|
|
|
7489
7716
|
@classmethod
|
|
7490
7717
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7606,12 +7833,19 @@ def FullyConnectedOptionsEnd(builder):
|
|
|
7606
7833
|
class FullyConnectedOptionsT(object):
|
|
7607
7834
|
|
|
7608
7835
|
# FullyConnectedOptionsT
|
|
7609
|
-
def __init__(
|
|
7610
|
-
self
|
|
7611
|
-
|
|
7612
|
-
|
|
7613
|
-
|
|
7614
|
-
|
|
7836
|
+
def __init__(
|
|
7837
|
+
self,
|
|
7838
|
+
fusedActivationFunction = 0,
|
|
7839
|
+
weightsFormat = 0,
|
|
7840
|
+
keepNumDims = False,
|
|
7841
|
+
asymmetricQuantizeInputs = False,
|
|
7842
|
+
quantizedBiasType = 0,
|
|
7843
|
+
):
|
|
7844
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7845
|
+
self.weightsFormat = weightsFormat # type: int
|
|
7846
|
+
self.keepNumDims = keepNumDims # type: bool
|
|
7847
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
7848
|
+
self.quantizedBiasType = quantizedBiasType # type: int
|
|
7615
7849
|
|
|
7616
7850
|
@classmethod
|
|
7617
7851
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7695,8 +7929,11 @@ def SoftmaxOptionsEnd(builder):
|
|
|
7695
7929
|
class SoftmaxOptionsT(object):
|
|
7696
7930
|
|
|
7697
7931
|
# SoftmaxOptionsT
|
|
7698
|
-
def __init__(
|
|
7699
|
-
self
|
|
7932
|
+
def __init__(
|
|
7933
|
+
self,
|
|
7934
|
+
beta = 0.0,
|
|
7935
|
+
):
|
|
7936
|
+
self.beta = beta # type: float
|
|
7700
7937
|
|
|
7701
7938
|
@classmethod
|
|
7702
7939
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7782,9 +8019,13 @@ def ConcatenationOptionsEnd(builder):
|
|
|
7782
8019
|
class ConcatenationOptionsT(object):
|
|
7783
8020
|
|
|
7784
8021
|
# ConcatenationOptionsT
|
|
7785
|
-
def __init__(
|
|
7786
|
-
self
|
|
7787
|
-
|
|
8022
|
+
def __init__(
|
|
8023
|
+
self,
|
|
8024
|
+
axis = 0,
|
|
8025
|
+
fusedActivationFunction = 0,
|
|
8026
|
+
):
|
|
8027
|
+
self.axis = axis # type: int
|
|
8028
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7788
8029
|
|
|
7789
8030
|
@classmethod
|
|
7790
8031
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7872,9 +8113,13 @@ def AddOptionsEnd(builder):
|
|
|
7872
8113
|
class AddOptionsT(object):
|
|
7873
8114
|
|
|
7874
8115
|
# AddOptionsT
|
|
7875
|
-
def __init__(
|
|
7876
|
-
self
|
|
7877
|
-
|
|
8116
|
+
def __init__(
|
|
8117
|
+
self,
|
|
8118
|
+
fusedActivationFunction = 0,
|
|
8119
|
+
potScaleInt16 = True,
|
|
8120
|
+
):
|
|
8121
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
8122
|
+
self.potScaleInt16 = potScaleInt16 # type: bool
|
|
7878
8123
|
|
|
7879
8124
|
@classmethod
|
|
7880
8125
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -7952,8 +8197,11 @@ def MulOptionsEnd(builder):
|
|
|
7952
8197
|
class MulOptionsT(object):
|
|
7953
8198
|
|
|
7954
8199
|
# MulOptionsT
|
|
7955
|
-
def __init__(
|
|
7956
|
-
self
|
|
8200
|
+
def __init__(
|
|
8201
|
+
self,
|
|
8202
|
+
fusedActivationFunction = 0,
|
|
8203
|
+
):
|
|
8204
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
7957
8205
|
|
|
7958
8206
|
@classmethod
|
|
7959
8207
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8029,8 +8277,11 @@ def L2NormOptionsEnd(builder):
|
|
|
8029
8277
|
class L2NormOptionsT(object):
|
|
8030
8278
|
|
|
8031
8279
|
# L2NormOptionsT
|
|
8032
|
-
def __init__(
|
|
8033
|
-
self
|
|
8280
|
+
def __init__(
|
|
8281
|
+
self,
|
|
8282
|
+
fusedActivationFunction = 0,
|
|
8283
|
+
):
|
|
8284
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
8034
8285
|
|
|
8035
8286
|
@classmethod
|
|
8036
8287
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8136,11 +8387,17 @@ def LocalResponseNormalizationOptionsEnd(builder):
|
|
|
8136
8387
|
class LocalResponseNormalizationOptionsT(object):
|
|
8137
8388
|
|
|
8138
8389
|
# LocalResponseNormalizationOptionsT
|
|
8139
|
-
def __init__(
|
|
8140
|
-
self
|
|
8141
|
-
|
|
8142
|
-
|
|
8143
|
-
|
|
8390
|
+
def __init__(
|
|
8391
|
+
self,
|
|
8392
|
+
radius = 0,
|
|
8393
|
+
bias = 0.0,
|
|
8394
|
+
alpha = 0.0,
|
|
8395
|
+
beta = 0.0,
|
|
8396
|
+
):
|
|
8397
|
+
self.radius = radius # type: int
|
|
8398
|
+
self.bias = bias # type: float
|
|
8399
|
+
self.alpha = alpha # type: float
|
|
8400
|
+
self.beta = beta # type: float
|
|
8144
8401
|
|
|
8145
8402
|
@classmethod
|
|
8146
8403
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8262,12 +8519,19 @@ def LSTMOptionsEnd(builder):
|
|
|
8262
8519
|
class LSTMOptionsT(object):
|
|
8263
8520
|
|
|
8264
8521
|
# LSTMOptionsT
|
|
8265
|
-
def __init__(
|
|
8266
|
-
self
|
|
8267
|
-
|
|
8268
|
-
|
|
8269
|
-
|
|
8270
|
-
|
|
8522
|
+
def __init__(
|
|
8523
|
+
self,
|
|
8524
|
+
fusedActivationFunction = 0,
|
|
8525
|
+
cellClip = 0.0,
|
|
8526
|
+
projClip = 0.0,
|
|
8527
|
+
kernelType = 0,
|
|
8528
|
+
asymmetricQuantizeInputs = False,
|
|
8529
|
+
):
|
|
8530
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
8531
|
+
self.cellClip = cellClip # type: float
|
|
8532
|
+
self.projClip = projClip # type: float
|
|
8533
|
+
self.kernelType = kernelType # type: int
|
|
8534
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
8271
8535
|
|
|
8272
8536
|
@classmethod
|
|
8273
8537
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8401,13 +8665,21 @@ def UnidirectionalSequenceLSTMOptionsEnd(builder):
|
|
|
8401
8665
|
class UnidirectionalSequenceLSTMOptionsT(object):
|
|
8402
8666
|
|
|
8403
8667
|
# UnidirectionalSequenceLSTMOptionsT
|
|
8404
|
-
def __init__(
|
|
8405
|
-
self
|
|
8406
|
-
|
|
8407
|
-
|
|
8408
|
-
|
|
8409
|
-
|
|
8410
|
-
|
|
8668
|
+
def __init__(
|
|
8669
|
+
self,
|
|
8670
|
+
fusedActivationFunction = 0,
|
|
8671
|
+
cellClip = 0.0,
|
|
8672
|
+
projClip = 0.0,
|
|
8673
|
+
timeMajor = False,
|
|
8674
|
+
asymmetricQuantizeInputs = False,
|
|
8675
|
+
diagonalRecurrentTensors = False,
|
|
8676
|
+
):
|
|
8677
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
8678
|
+
self.cellClip = cellClip # type: float
|
|
8679
|
+
self.projClip = projClip # type: float
|
|
8680
|
+
self.timeMajor = timeMajor # type: bool
|
|
8681
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
8682
|
+
self.diagonalRecurrentTensors = diagonalRecurrentTensors # type: bool
|
|
8411
8683
|
|
|
8412
8684
|
@classmethod
|
|
8413
8685
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8543,13 +8815,21 @@ def BidirectionalSequenceLSTMOptionsEnd(builder):
|
|
|
8543
8815
|
class BidirectionalSequenceLSTMOptionsT(object):
|
|
8544
8816
|
|
|
8545
8817
|
# BidirectionalSequenceLSTMOptionsT
|
|
8546
|
-
def __init__(
|
|
8547
|
-
self
|
|
8548
|
-
|
|
8549
|
-
|
|
8550
|
-
|
|
8551
|
-
|
|
8552
|
-
|
|
8818
|
+
def __init__(
|
|
8819
|
+
self,
|
|
8820
|
+
fusedActivationFunction = 0,
|
|
8821
|
+
cellClip = 0.0,
|
|
8822
|
+
projClip = 0.0,
|
|
8823
|
+
mergeOutputs = False,
|
|
8824
|
+
timeMajor = True,
|
|
8825
|
+
asymmetricQuantizeInputs = False,
|
|
8826
|
+
):
|
|
8827
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
8828
|
+
self.cellClip = cellClip # type: float
|
|
8829
|
+
self.projClip = projClip # type: float
|
|
8830
|
+
self.mergeOutputs = mergeOutputs # type: bool
|
|
8831
|
+
self.timeMajor = timeMajor # type: bool
|
|
8832
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
8553
8833
|
|
|
8554
8834
|
@classmethod
|
|
8555
8835
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8645,9 +8925,13 @@ def ResizeBilinearOptionsEnd(builder):
|
|
|
8645
8925
|
class ResizeBilinearOptionsT(object):
|
|
8646
8926
|
|
|
8647
8927
|
# ResizeBilinearOptionsT
|
|
8648
|
-
def __init__(
|
|
8649
|
-
self
|
|
8650
|
-
|
|
8928
|
+
def __init__(
|
|
8929
|
+
self,
|
|
8930
|
+
alignCorners = False,
|
|
8931
|
+
halfPixelCenters = False,
|
|
8932
|
+
):
|
|
8933
|
+
self.alignCorners = alignCorners # type: bool
|
|
8934
|
+
self.halfPixelCenters = halfPixelCenters # type: bool
|
|
8651
8935
|
|
|
8652
8936
|
@classmethod
|
|
8653
8937
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8735,9 +9019,13 @@ def ResizeNearestNeighborOptionsEnd(builder):
|
|
|
8735
9019
|
class ResizeNearestNeighborOptionsT(object):
|
|
8736
9020
|
|
|
8737
9021
|
# ResizeNearestNeighborOptionsT
|
|
8738
|
-
def __init__(
|
|
8739
|
-
self
|
|
8740
|
-
|
|
9022
|
+
def __init__(
|
|
9023
|
+
self,
|
|
9024
|
+
alignCorners = False,
|
|
9025
|
+
halfPixelCenters = False,
|
|
9026
|
+
):
|
|
9027
|
+
self.alignCorners = alignCorners # type: bool
|
|
9028
|
+
self.halfPixelCenters = halfPixelCenters # type: bool
|
|
8741
9029
|
|
|
8742
9030
|
@classmethod
|
|
8743
9031
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8815,8 +9103,11 @@ def CallOptionsEnd(builder):
|
|
|
8815
9103
|
class CallOptionsT(object):
|
|
8816
9104
|
|
|
8817
9105
|
# CallOptionsT
|
|
8818
|
-
def __init__(
|
|
8819
|
-
self
|
|
9106
|
+
def __init__(
|
|
9107
|
+
self,
|
|
9108
|
+
subgraph = 0,
|
|
9109
|
+
):
|
|
9110
|
+
self.subgraph = subgraph # type: int
|
|
8820
9111
|
|
|
8821
9112
|
@classmethod
|
|
8822
9113
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -8882,7 +9173,9 @@ def PadOptionsEnd(builder):
|
|
|
8882
9173
|
class PadOptionsT(object):
|
|
8883
9174
|
|
|
8884
9175
|
# PadOptionsT
|
|
8885
|
-
def __init__(
|
|
9176
|
+
def __init__(
|
|
9177
|
+
self,
|
|
9178
|
+
):
|
|
8886
9179
|
pass
|
|
8887
9180
|
|
|
8888
9181
|
@classmethod
|
|
@@ -8947,7 +9240,9 @@ def PadV2OptionsEnd(builder):
|
|
|
8947
9240
|
class PadV2OptionsT(object):
|
|
8948
9241
|
|
|
8949
9242
|
# PadV2OptionsT
|
|
8950
|
-
def __init__(
|
|
9243
|
+
def __init__(
|
|
9244
|
+
self,
|
|
9245
|
+
):
|
|
8951
9246
|
pass
|
|
8952
9247
|
|
|
8953
9248
|
@classmethod
|
|
@@ -9049,8 +9344,11 @@ except:
|
|
|
9049
9344
|
class ReshapeOptionsT(object):
|
|
9050
9345
|
|
|
9051
9346
|
# ReshapeOptionsT
|
|
9052
|
-
def __init__(
|
|
9053
|
-
self
|
|
9347
|
+
def __init__(
|
|
9348
|
+
self,
|
|
9349
|
+
newShape = None,
|
|
9350
|
+
):
|
|
9351
|
+
self.newShape = newShape # type: Optional[List[int]]
|
|
9054
9352
|
|
|
9055
9353
|
@classmethod
|
|
9056
9354
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9131,7 +9429,9 @@ def SpaceToBatchNDOptionsEnd(builder):
|
|
|
9131
9429
|
class SpaceToBatchNDOptionsT(object):
|
|
9132
9430
|
|
|
9133
9431
|
# SpaceToBatchNDOptionsT
|
|
9134
|
-
def __init__(
|
|
9432
|
+
def __init__(
|
|
9433
|
+
self,
|
|
9434
|
+
):
|
|
9135
9435
|
pass
|
|
9136
9436
|
|
|
9137
9437
|
@classmethod
|
|
@@ -9196,7 +9496,9 @@ def BatchToSpaceNDOptionsEnd(builder):
|
|
|
9196
9496
|
class BatchToSpaceNDOptionsT(object):
|
|
9197
9497
|
|
|
9198
9498
|
# BatchToSpaceNDOptionsT
|
|
9199
|
-
def __init__(
|
|
9499
|
+
def __init__(
|
|
9500
|
+
self,
|
|
9501
|
+
):
|
|
9200
9502
|
pass
|
|
9201
9503
|
|
|
9202
9504
|
@classmethod
|
|
@@ -9291,10 +9593,15 @@ def SkipGramOptionsEnd(builder):
|
|
|
9291
9593
|
class SkipGramOptionsT(object):
|
|
9292
9594
|
|
|
9293
9595
|
# SkipGramOptionsT
|
|
9294
|
-
def __init__(
|
|
9295
|
-
self
|
|
9296
|
-
|
|
9297
|
-
|
|
9596
|
+
def __init__(
|
|
9597
|
+
self,
|
|
9598
|
+
ngramSize = 0,
|
|
9599
|
+
maxSkipSize = 0,
|
|
9600
|
+
includeAllNgrams = False,
|
|
9601
|
+
):
|
|
9602
|
+
self.ngramSize = ngramSize # type: int
|
|
9603
|
+
self.maxSkipSize = maxSkipSize # type: int
|
|
9604
|
+
self.includeAllNgrams = includeAllNgrams # type: bool
|
|
9298
9605
|
|
|
9299
9606
|
@classmethod
|
|
9300
9607
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9374,8 +9681,11 @@ def SpaceToDepthOptionsEnd(builder):
|
|
|
9374
9681
|
class SpaceToDepthOptionsT(object):
|
|
9375
9682
|
|
|
9376
9683
|
# SpaceToDepthOptionsT
|
|
9377
|
-
def __init__(
|
|
9378
|
-
self
|
|
9684
|
+
def __init__(
|
|
9685
|
+
self,
|
|
9686
|
+
blockSize = 0,
|
|
9687
|
+
):
|
|
9688
|
+
self.blockSize = blockSize # type: int
|
|
9379
9689
|
|
|
9380
9690
|
@classmethod
|
|
9381
9691
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9451,8 +9761,11 @@ def DepthToSpaceOptionsEnd(builder):
|
|
|
9451
9761
|
class DepthToSpaceOptionsT(object):
|
|
9452
9762
|
|
|
9453
9763
|
# DepthToSpaceOptionsT
|
|
9454
|
-
def __init__(
|
|
9455
|
-
self
|
|
9764
|
+
def __init__(
|
|
9765
|
+
self,
|
|
9766
|
+
blockSize = 0,
|
|
9767
|
+
):
|
|
9768
|
+
self.blockSize = blockSize # type: int
|
|
9456
9769
|
|
|
9457
9770
|
@classmethod
|
|
9458
9771
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9538,9 +9851,13 @@ def SubOptionsEnd(builder):
|
|
|
9538
9851
|
class SubOptionsT(object):
|
|
9539
9852
|
|
|
9540
9853
|
# SubOptionsT
|
|
9541
|
-
def __init__(
|
|
9542
|
-
self
|
|
9543
|
-
|
|
9854
|
+
def __init__(
|
|
9855
|
+
self,
|
|
9856
|
+
fusedActivationFunction = 0,
|
|
9857
|
+
potScaleInt16 = True,
|
|
9858
|
+
):
|
|
9859
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
9860
|
+
self.potScaleInt16 = potScaleInt16 # type: bool
|
|
9544
9861
|
|
|
9545
9862
|
@classmethod
|
|
9546
9863
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9618,8 +9935,11 @@ def DivOptionsEnd(builder):
|
|
|
9618
9935
|
class DivOptionsT(object):
|
|
9619
9936
|
|
|
9620
9937
|
# DivOptionsT
|
|
9621
|
-
def __init__(
|
|
9622
|
-
self
|
|
9938
|
+
def __init__(
|
|
9939
|
+
self,
|
|
9940
|
+
fusedActivationFunction = 0,
|
|
9941
|
+
):
|
|
9942
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
9623
9943
|
|
|
9624
9944
|
@classmethod
|
|
9625
9945
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9685,7 +10005,9 @@ def TopKV2OptionsEnd(builder):
|
|
|
9685
10005
|
class TopKV2OptionsT(object):
|
|
9686
10006
|
|
|
9687
10007
|
# TopKV2OptionsT
|
|
9688
|
-
def __init__(
|
|
10008
|
+
def __init__(
|
|
10009
|
+
self,
|
|
10010
|
+
):
|
|
9689
10011
|
pass
|
|
9690
10012
|
|
|
9691
10013
|
@classmethod
|
|
@@ -9760,8 +10082,11 @@ def EmbeddingLookupSparseOptionsEnd(builder):
|
|
|
9760
10082
|
class EmbeddingLookupSparseOptionsT(object):
|
|
9761
10083
|
|
|
9762
10084
|
# EmbeddingLookupSparseOptionsT
|
|
9763
|
-
def __init__(
|
|
9764
|
-
self
|
|
10085
|
+
def __init__(
|
|
10086
|
+
self,
|
|
10087
|
+
combiner = 0,
|
|
10088
|
+
):
|
|
10089
|
+
self.combiner = combiner # type: int
|
|
9765
10090
|
|
|
9766
10091
|
@classmethod
|
|
9767
10092
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9847,9 +10172,13 @@ def GatherOptionsEnd(builder):
|
|
|
9847
10172
|
class GatherOptionsT(object):
|
|
9848
10173
|
|
|
9849
10174
|
# GatherOptionsT
|
|
9850
|
-
def __init__(
|
|
9851
|
-
self
|
|
9852
|
-
|
|
10175
|
+
def __init__(
|
|
10176
|
+
self,
|
|
10177
|
+
axis = 0,
|
|
10178
|
+
batchDims = 0,
|
|
10179
|
+
):
|
|
10180
|
+
self.axis = axis # type: int
|
|
10181
|
+
self.batchDims = batchDims # type: int
|
|
9853
10182
|
|
|
9854
10183
|
@classmethod
|
|
9855
10184
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -9917,7 +10246,9 @@ def TransposeOptionsEnd(builder):
|
|
|
9917
10246
|
class TransposeOptionsT(object):
|
|
9918
10247
|
|
|
9919
10248
|
# TransposeOptionsT
|
|
9920
|
-
def __init__(
|
|
10249
|
+
def __init__(
|
|
10250
|
+
self,
|
|
10251
|
+
):
|
|
9921
10252
|
pass
|
|
9922
10253
|
|
|
9923
10254
|
@classmethod
|
|
@@ -9982,7 +10313,9 @@ def ExpOptionsEnd(builder):
|
|
|
9982
10313
|
class ExpOptionsT(object):
|
|
9983
10314
|
|
|
9984
10315
|
# ExpOptionsT
|
|
9985
|
-
def __init__(
|
|
10316
|
+
def __init__(
|
|
10317
|
+
self,
|
|
10318
|
+
):
|
|
9986
10319
|
pass
|
|
9987
10320
|
|
|
9988
10321
|
@classmethod
|
|
@@ -10047,7 +10380,9 @@ def CosOptionsEnd(builder):
|
|
|
10047
10380
|
class CosOptionsT(object):
|
|
10048
10381
|
|
|
10049
10382
|
# CosOptionsT
|
|
10050
|
-
def __init__(
|
|
10383
|
+
def __init__(
|
|
10384
|
+
self,
|
|
10385
|
+
):
|
|
10051
10386
|
pass
|
|
10052
10387
|
|
|
10053
10388
|
@classmethod
|
|
@@ -10122,8 +10457,11 @@ def ReducerOptionsEnd(builder):
|
|
|
10122
10457
|
class ReducerOptionsT(object):
|
|
10123
10458
|
|
|
10124
10459
|
# ReducerOptionsT
|
|
10125
|
-
def __init__(
|
|
10126
|
-
self
|
|
10460
|
+
def __init__(
|
|
10461
|
+
self,
|
|
10462
|
+
keepDims = False,
|
|
10463
|
+
):
|
|
10464
|
+
self.keepDims = keepDims # type: bool
|
|
10127
10465
|
|
|
10128
10466
|
@classmethod
|
|
10129
10467
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10226,8 +10564,11 @@ except:
|
|
|
10226
10564
|
class SqueezeOptionsT(object):
|
|
10227
10565
|
|
|
10228
10566
|
# SqueezeOptionsT
|
|
10229
|
-
def __init__(
|
|
10230
|
-
self
|
|
10567
|
+
def __init__(
|
|
10568
|
+
self,
|
|
10569
|
+
squeezeDims = None,
|
|
10570
|
+
):
|
|
10571
|
+
self.squeezeDims = squeezeDims # type: Optional[List[int]]
|
|
10231
10572
|
|
|
10232
10573
|
@classmethod
|
|
10233
10574
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10318,8 +10659,11 @@ def SplitOptionsEnd(builder):
|
|
|
10318
10659
|
class SplitOptionsT(object):
|
|
10319
10660
|
|
|
10320
10661
|
# SplitOptionsT
|
|
10321
|
-
def __init__(
|
|
10322
|
-
self
|
|
10662
|
+
def __init__(
|
|
10663
|
+
self,
|
|
10664
|
+
numSplits = 0,
|
|
10665
|
+
):
|
|
10666
|
+
self.numSplits = numSplits # type: int
|
|
10323
10667
|
|
|
10324
10668
|
@classmethod
|
|
10325
10669
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10395,8 +10739,11 @@ def SplitVOptionsEnd(builder):
|
|
|
10395
10739
|
class SplitVOptionsT(object):
|
|
10396
10740
|
|
|
10397
10741
|
# SplitVOptionsT
|
|
10398
|
-
def __init__(
|
|
10399
|
-
self
|
|
10742
|
+
def __init__(
|
|
10743
|
+
self,
|
|
10744
|
+
numSplits = 0,
|
|
10745
|
+
):
|
|
10746
|
+
self.numSplits = numSplits # type: int
|
|
10400
10747
|
|
|
10401
10748
|
@classmethod
|
|
10402
10749
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10522,13 +10869,21 @@ def StridedSliceOptionsEnd(builder):
|
|
|
10522
10869
|
class StridedSliceOptionsT(object):
|
|
10523
10870
|
|
|
10524
10871
|
# StridedSliceOptionsT
|
|
10525
|
-
def __init__(
|
|
10526
|
-
self
|
|
10527
|
-
|
|
10528
|
-
|
|
10529
|
-
|
|
10530
|
-
|
|
10531
|
-
|
|
10872
|
+
def __init__(
|
|
10873
|
+
self,
|
|
10874
|
+
beginMask = 0,
|
|
10875
|
+
endMask = 0,
|
|
10876
|
+
ellipsisMask = 0,
|
|
10877
|
+
newAxisMask = 0,
|
|
10878
|
+
shrinkAxisMask = 0,
|
|
10879
|
+
offset = False,
|
|
10880
|
+
):
|
|
10881
|
+
self.beginMask = beginMask # type: int
|
|
10882
|
+
self.endMask = endMask # type: int
|
|
10883
|
+
self.ellipsisMask = ellipsisMask # type: int
|
|
10884
|
+
self.newAxisMask = newAxisMask # type: int
|
|
10885
|
+
self.shrinkAxisMask = shrinkAxisMask # type: int
|
|
10886
|
+
self.offset = offset # type: bool
|
|
10532
10887
|
|
|
10533
10888
|
@classmethod
|
|
10534
10889
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10604,7 +10959,9 @@ def LogSoftmaxOptionsEnd(builder):
|
|
|
10604
10959
|
class LogSoftmaxOptionsT(object):
|
|
10605
10960
|
|
|
10606
10961
|
# LogSoftmaxOptionsT
|
|
10607
|
-
def __init__(
|
|
10962
|
+
def __init__(
|
|
10963
|
+
self,
|
|
10964
|
+
):
|
|
10608
10965
|
pass
|
|
10609
10966
|
|
|
10610
10967
|
@classmethod
|
|
@@ -10689,9 +11046,13 @@ def CastOptionsEnd(builder):
|
|
|
10689
11046
|
class CastOptionsT(object):
|
|
10690
11047
|
|
|
10691
11048
|
# CastOptionsT
|
|
10692
|
-
def __init__(
|
|
10693
|
-
self
|
|
10694
|
-
|
|
11049
|
+
def __init__(
|
|
11050
|
+
self,
|
|
11051
|
+
inDataType = 0,
|
|
11052
|
+
outDataType = 0,
|
|
11053
|
+
):
|
|
11054
|
+
self.inDataType = inDataType # type: int
|
|
11055
|
+
self.outDataType = outDataType # type: int
|
|
10695
11056
|
|
|
10696
11057
|
@classmethod
|
|
10697
11058
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -10759,7 +11120,9 @@ def DequantizeOptionsEnd(builder):
|
|
|
10759
11120
|
class DequantizeOptionsT(object):
|
|
10760
11121
|
|
|
10761
11122
|
# DequantizeOptionsT
|
|
10762
|
-
def __init__(
|
|
11123
|
+
def __init__(
|
|
11124
|
+
self,
|
|
11125
|
+
):
|
|
10763
11126
|
pass
|
|
10764
11127
|
|
|
10765
11128
|
@classmethod
|
|
@@ -10824,7 +11187,9 @@ def MaximumMinimumOptionsEnd(builder):
|
|
|
10824
11187
|
class MaximumMinimumOptionsT(object):
|
|
10825
11188
|
|
|
10826
11189
|
# MaximumMinimumOptionsT
|
|
10827
|
-
def __init__(
|
|
11190
|
+
def __init__(
|
|
11191
|
+
self,
|
|
11192
|
+
):
|
|
10828
11193
|
pass
|
|
10829
11194
|
|
|
10830
11195
|
@classmethod
|
|
@@ -10889,7 +11254,9 @@ def TileOptionsEnd(builder):
|
|
|
10889
11254
|
class TileOptionsT(object):
|
|
10890
11255
|
|
|
10891
11256
|
# TileOptionsT
|
|
10892
|
-
def __init__(
|
|
11257
|
+
def __init__(
|
|
11258
|
+
self,
|
|
11259
|
+
):
|
|
10893
11260
|
pass
|
|
10894
11261
|
|
|
10895
11262
|
@classmethod
|
|
@@ -10964,8 +11331,11 @@ def ArgMaxOptionsEnd(builder):
|
|
|
10964
11331
|
class ArgMaxOptionsT(object):
|
|
10965
11332
|
|
|
10966
11333
|
# ArgMaxOptionsT
|
|
10967
|
-
def __init__(
|
|
10968
|
-
self
|
|
11334
|
+
def __init__(
|
|
11335
|
+
self,
|
|
11336
|
+
outputType = 0,
|
|
11337
|
+
):
|
|
11338
|
+
self.outputType = outputType # type: int
|
|
10969
11339
|
|
|
10970
11340
|
@classmethod
|
|
10971
11341
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -11041,8 +11411,11 @@ def ArgMinOptionsEnd(builder):
|
|
|
11041
11411
|
class ArgMinOptionsT(object):
|
|
11042
11412
|
|
|
11043
11413
|
# ArgMinOptionsT
|
|
11044
|
-
def __init__(
|
|
11045
|
-
self
|
|
11414
|
+
def __init__(
|
|
11415
|
+
self,
|
|
11416
|
+
outputType = 0,
|
|
11417
|
+
):
|
|
11418
|
+
self.outputType = outputType # type: int
|
|
11046
11419
|
|
|
11047
11420
|
@classmethod
|
|
11048
11421
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -11108,7 +11481,9 @@ def GreaterOptionsEnd(builder):
|
|
|
11108
11481
|
class GreaterOptionsT(object):
|
|
11109
11482
|
|
|
11110
11483
|
# GreaterOptionsT
|
|
11111
|
-
def __init__(
|
|
11484
|
+
def __init__(
|
|
11485
|
+
self,
|
|
11486
|
+
):
|
|
11112
11487
|
pass
|
|
11113
11488
|
|
|
11114
11489
|
@classmethod
|
|
@@ -11173,7 +11548,9 @@ def GreaterEqualOptionsEnd(builder):
|
|
|
11173
11548
|
class GreaterEqualOptionsT(object):
|
|
11174
11549
|
|
|
11175
11550
|
# GreaterEqualOptionsT
|
|
11176
|
-
def __init__(
|
|
11551
|
+
def __init__(
|
|
11552
|
+
self,
|
|
11553
|
+
):
|
|
11177
11554
|
pass
|
|
11178
11555
|
|
|
11179
11556
|
@classmethod
|
|
@@ -11238,7 +11615,9 @@ def LessOptionsEnd(builder):
|
|
|
11238
11615
|
class LessOptionsT(object):
|
|
11239
11616
|
|
|
11240
11617
|
# LessOptionsT
|
|
11241
|
-
def __init__(
|
|
11618
|
+
def __init__(
|
|
11619
|
+
self,
|
|
11620
|
+
):
|
|
11242
11621
|
pass
|
|
11243
11622
|
|
|
11244
11623
|
@classmethod
|
|
@@ -11303,7 +11682,9 @@ def LessEqualOptionsEnd(builder):
|
|
|
11303
11682
|
class LessEqualOptionsT(object):
|
|
11304
11683
|
|
|
11305
11684
|
# LessEqualOptionsT
|
|
11306
|
-
def __init__(
|
|
11685
|
+
def __init__(
|
|
11686
|
+
self,
|
|
11687
|
+
):
|
|
11307
11688
|
pass
|
|
11308
11689
|
|
|
11309
11690
|
@classmethod
|
|
@@ -11368,7 +11749,9 @@ def NegOptionsEnd(builder):
|
|
|
11368
11749
|
class NegOptionsT(object):
|
|
11369
11750
|
|
|
11370
11751
|
# NegOptionsT
|
|
11371
|
-
def __init__(
|
|
11752
|
+
def __init__(
|
|
11753
|
+
self,
|
|
11754
|
+
):
|
|
11372
11755
|
pass
|
|
11373
11756
|
|
|
11374
11757
|
@classmethod
|
|
@@ -11433,7 +11816,9 @@ def SelectOptionsEnd(builder):
|
|
|
11433
11816
|
class SelectOptionsT(object):
|
|
11434
11817
|
|
|
11435
11818
|
# SelectOptionsT
|
|
11436
|
-
def __init__(
|
|
11819
|
+
def __init__(
|
|
11820
|
+
self,
|
|
11821
|
+
):
|
|
11437
11822
|
pass
|
|
11438
11823
|
|
|
11439
11824
|
@classmethod
|
|
@@ -11498,7 +11883,9 @@ def SliceOptionsEnd(builder):
|
|
|
11498
11883
|
class SliceOptionsT(object):
|
|
11499
11884
|
|
|
11500
11885
|
# SliceOptionsT
|
|
11501
|
-
def __init__(
|
|
11886
|
+
def __init__(
|
|
11887
|
+
self,
|
|
11888
|
+
):
|
|
11502
11889
|
pass
|
|
11503
11890
|
|
|
11504
11891
|
@classmethod
|
|
@@ -11613,12 +12000,19 @@ def TransposeConvOptionsEnd(builder):
|
|
|
11613
12000
|
class TransposeConvOptionsT(object):
|
|
11614
12001
|
|
|
11615
12002
|
# TransposeConvOptionsT
|
|
11616
|
-
def __init__(
|
|
11617
|
-
self
|
|
11618
|
-
|
|
11619
|
-
|
|
11620
|
-
|
|
11621
|
-
|
|
12003
|
+
def __init__(
|
|
12004
|
+
self,
|
|
12005
|
+
padding = 0,
|
|
12006
|
+
strideW = 0,
|
|
12007
|
+
strideH = 0,
|
|
12008
|
+
fusedActivationFunction = 0,
|
|
12009
|
+
quantizedBiasType = 0,
|
|
12010
|
+
):
|
|
12011
|
+
self.padding = padding # type: int
|
|
12012
|
+
self.strideW = strideW # type: int
|
|
12013
|
+
self.strideH = strideH # type: int
|
|
12014
|
+
self.fusedActivationFunction = fusedActivationFunction # type: int
|
|
12015
|
+
self.quantizedBiasType = quantizedBiasType # type: int
|
|
11622
12016
|
|
|
11623
12017
|
@classmethod
|
|
11624
12018
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -11692,7 +12086,9 @@ def ExpandDimsOptionsEnd(builder):
|
|
|
11692
12086
|
class ExpandDimsOptionsT(object):
|
|
11693
12087
|
|
|
11694
12088
|
# ExpandDimsOptionsT
|
|
11695
|
-
def __init__(
|
|
12089
|
+
def __init__(
|
|
12090
|
+
self,
|
|
12091
|
+
):
|
|
11696
12092
|
pass
|
|
11697
12093
|
|
|
11698
12094
|
@classmethod
|
|
@@ -11767,8 +12163,11 @@ def SparseToDenseOptionsEnd(builder):
|
|
|
11767
12163
|
class SparseToDenseOptionsT(object):
|
|
11768
12164
|
|
|
11769
12165
|
# SparseToDenseOptionsT
|
|
11770
|
-
def __init__(
|
|
11771
|
-
self
|
|
12166
|
+
def __init__(
|
|
12167
|
+
self,
|
|
12168
|
+
validateIndices = False,
|
|
12169
|
+
):
|
|
12170
|
+
self.validateIndices = validateIndices # type: bool
|
|
11772
12171
|
|
|
11773
12172
|
@classmethod
|
|
11774
12173
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -11834,7 +12233,9 @@ def EqualOptionsEnd(builder):
|
|
|
11834
12233
|
class EqualOptionsT(object):
|
|
11835
12234
|
|
|
11836
12235
|
# EqualOptionsT
|
|
11837
|
-
def __init__(
|
|
12236
|
+
def __init__(
|
|
12237
|
+
self,
|
|
12238
|
+
):
|
|
11838
12239
|
pass
|
|
11839
12240
|
|
|
11840
12241
|
@classmethod
|
|
@@ -11899,7 +12300,9 @@ def NotEqualOptionsEnd(builder):
|
|
|
11899
12300
|
class NotEqualOptionsT(object):
|
|
11900
12301
|
|
|
11901
12302
|
# NotEqualOptionsT
|
|
11902
|
-
def __init__(
|
|
12303
|
+
def __init__(
|
|
12304
|
+
self,
|
|
12305
|
+
):
|
|
11903
12306
|
pass
|
|
11904
12307
|
|
|
11905
12308
|
@classmethod
|
|
@@ -11974,8 +12377,11 @@ def ShapeOptionsEnd(builder):
|
|
|
11974
12377
|
class ShapeOptionsT(object):
|
|
11975
12378
|
|
|
11976
12379
|
# ShapeOptionsT
|
|
11977
|
-
def __init__(
|
|
11978
|
-
self
|
|
12380
|
+
def __init__(
|
|
12381
|
+
self,
|
|
12382
|
+
outType = 0,
|
|
12383
|
+
):
|
|
12384
|
+
self.outType = outType # type: int
|
|
11979
12385
|
|
|
11980
12386
|
@classmethod
|
|
11981
12387
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -12041,7 +12447,9 @@ def RankOptionsEnd(builder):
|
|
|
12041
12447
|
class RankOptionsT(object):
|
|
12042
12448
|
|
|
12043
12449
|
# RankOptionsT
|
|
12044
|
-
def __init__(
|
|
12450
|
+
def __init__(
|
|
12451
|
+
self,
|
|
12452
|
+
):
|
|
12045
12453
|
pass
|
|
12046
12454
|
|
|
12047
12455
|
@classmethod
|
|
@@ -12106,7 +12514,9 @@ def PowOptionsEnd(builder):
|
|
|
12106
12514
|
class PowOptionsT(object):
|
|
12107
12515
|
|
|
12108
12516
|
# PowOptionsT
|
|
12109
|
-
def __init__(
|
|
12517
|
+
def __init__(
|
|
12518
|
+
self,
|
|
12519
|
+
):
|
|
12110
12520
|
pass
|
|
12111
12521
|
|
|
12112
12522
|
@classmethod
|
|
@@ -12211,11 +12621,17 @@ def FakeQuantOptionsEnd(builder):
|
|
|
12211
12621
|
class FakeQuantOptionsT(object):
|
|
12212
12622
|
|
|
12213
12623
|
# FakeQuantOptionsT
|
|
12214
|
-
def __init__(
|
|
12215
|
-
self
|
|
12216
|
-
|
|
12217
|
-
|
|
12218
|
-
|
|
12624
|
+
def __init__(
|
|
12625
|
+
self,
|
|
12626
|
+
min = 0.0,
|
|
12627
|
+
max = 0.0,
|
|
12628
|
+
numBits = 0,
|
|
12629
|
+
narrowRange = False,
|
|
12630
|
+
):
|
|
12631
|
+
self.min = min # type: float
|
|
12632
|
+
self.max = max # type: float
|
|
12633
|
+
self.numBits = numBits # type: int
|
|
12634
|
+
self.narrowRange = narrowRange # type: bool
|
|
12219
12635
|
|
|
12220
12636
|
@classmethod
|
|
12221
12637
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -12307,9 +12723,13 @@ def PackOptionsEnd(builder):
|
|
|
12307
12723
|
class PackOptionsT(object):
|
|
12308
12724
|
|
|
12309
12725
|
# PackOptionsT
|
|
12310
|
-
def __init__(
|
|
12311
|
-
self
|
|
12312
|
-
|
|
12726
|
+
def __init__(
|
|
12727
|
+
self,
|
|
12728
|
+
valuesCount = 0,
|
|
12729
|
+
axis = 0,
|
|
12730
|
+
):
|
|
12731
|
+
self.valuesCount = valuesCount # type: int
|
|
12732
|
+
self.axis = axis # type: int
|
|
12313
12733
|
|
|
12314
12734
|
@classmethod
|
|
12315
12735
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -12377,7 +12797,9 @@ def LogicalOrOptionsEnd(builder):
|
|
|
12377
12797
|
class LogicalOrOptionsT(object):
|
|
12378
12798
|
|
|
12379
12799
|
# LogicalOrOptionsT
|
|
12380
|
-
def __init__(
|
|
12800
|
+
def __init__(
|
|
12801
|
+
self,
|
|
12802
|
+
):
|
|
12381
12803
|
pass
|
|
12382
12804
|
|
|
12383
12805
|
@classmethod
|
|
@@ -12452,8 +12874,11 @@ def OneHotOptionsEnd(builder):
|
|
|
12452
12874
|
class OneHotOptionsT(object):
|
|
12453
12875
|
|
|
12454
12876
|
# OneHotOptionsT
|
|
12455
|
-
def __init__(
|
|
12456
|
-
self
|
|
12877
|
+
def __init__(
|
|
12878
|
+
self,
|
|
12879
|
+
axis = 0,
|
|
12880
|
+
):
|
|
12881
|
+
self.axis = axis # type: int
|
|
12457
12882
|
|
|
12458
12883
|
@classmethod
|
|
12459
12884
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -12519,7 +12944,9 @@ def AbsOptionsEnd(builder):
|
|
|
12519
12944
|
class AbsOptionsT(object):
|
|
12520
12945
|
|
|
12521
12946
|
# AbsOptionsT
|
|
12522
|
-
def __init__(
|
|
12947
|
+
def __init__(
|
|
12948
|
+
self,
|
|
12949
|
+
):
|
|
12523
12950
|
pass
|
|
12524
12951
|
|
|
12525
12952
|
@classmethod
|
|
@@ -12584,7 +13011,9 @@ def HardSwishOptionsEnd(builder):
|
|
|
12584
13011
|
class HardSwishOptionsT(object):
|
|
12585
13012
|
|
|
12586
13013
|
# HardSwishOptionsT
|
|
12587
|
-
def __init__(
|
|
13014
|
+
def __init__(
|
|
13015
|
+
self,
|
|
13016
|
+
):
|
|
12588
13017
|
pass
|
|
12589
13018
|
|
|
12590
13019
|
@classmethod
|
|
@@ -12649,7 +13078,9 @@ def LogicalAndOptionsEnd(builder):
|
|
|
12649
13078
|
class LogicalAndOptionsT(object):
|
|
12650
13079
|
|
|
12651
13080
|
# LogicalAndOptionsT
|
|
12652
|
-
def __init__(
|
|
13081
|
+
def __init__(
|
|
13082
|
+
self,
|
|
13083
|
+
):
|
|
12653
13084
|
pass
|
|
12654
13085
|
|
|
12655
13086
|
@classmethod
|
|
@@ -12714,7 +13145,9 @@ def LogicalNotOptionsEnd(builder):
|
|
|
12714
13145
|
class LogicalNotOptionsT(object):
|
|
12715
13146
|
|
|
12716
13147
|
# LogicalNotOptionsT
|
|
12717
|
-
def __init__(
|
|
13148
|
+
def __init__(
|
|
13149
|
+
self,
|
|
13150
|
+
):
|
|
12718
13151
|
pass
|
|
12719
13152
|
|
|
12720
13153
|
@classmethod
|
|
@@ -12799,9 +13232,13 @@ def UnpackOptionsEnd(builder):
|
|
|
12799
13232
|
class UnpackOptionsT(object):
|
|
12800
13233
|
|
|
12801
13234
|
# UnpackOptionsT
|
|
12802
|
-
def __init__(
|
|
12803
|
-
self
|
|
12804
|
-
|
|
13235
|
+
def __init__(
|
|
13236
|
+
self,
|
|
13237
|
+
num = 0,
|
|
13238
|
+
axis = 0,
|
|
13239
|
+
):
|
|
13240
|
+
self.num = num # type: int
|
|
13241
|
+
self.axis = axis # type: int
|
|
12805
13242
|
|
|
12806
13243
|
@classmethod
|
|
12807
13244
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -12869,7 +13306,9 @@ def FloorDivOptionsEnd(builder):
|
|
|
12869
13306
|
class FloorDivOptionsT(object):
|
|
12870
13307
|
|
|
12871
13308
|
# FloorDivOptionsT
|
|
12872
|
-
def __init__(
|
|
13309
|
+
def __init__(
|
|
13310
|
+
self,
|
|
13311
|
+
):
|
|
12873
13312
|
pass
|
|
12874
13313
|
|
|
12875
13314
|
@classmethod
|
|
@@ -12934,7 +13373,9 @@ def SquareOptionsEnd(builder):
|
|
|
12934
13373
|
class SquareOptionsT(object):
|
|
12935
13374
|
|
|
12936
13375
|
# SquareOptionsT
|
|
12937
|
-
def __init__(
|
|
13376
|
+
def __init__(
|
|
13377
|
+
self,
|
|
13378
|
+
):
|
|
12938
13379
|
pass
|
|
12939
13380
|
|
|
12940
13381
|
@classmethod
|
|
@@ -12999,7 +13440,9 @@ def ZerosLikeOptionsEnd(builder):
|
|
|
12999
13440
|
class ZerosLikeOptionsT(object):
|
|
13000
13441
|
|
|
13001
13442
|
# ZerosLikeOptionsT
|
|
13002
|
-
def __init__(
|
|
13443
|
+
def __init__(
|
|
13444
|
+
self,
|
|
13445
|
+
):
|
|
13003
13446
|
pass
|
|
13004
13447
|
|
|
13005
13448
|
@classmethod
|
|
@@ -13064,7 +13507,9 @@ def FillOptionsEnd(builder):
|
|
|
13064
13507
|
class FillOptionsT(object):
|
|
13065
13508
|
|
|
13066
13509
|
# FillOptionsT
|
|
13067
|
-
def __init__(
|
|
13510
|
+
def __init__(
|
|
13511
|
+
self,
|
|
13512
|
+
):
|
|
13068
13513
|
pass
|
|
13069
13514
|
|
|
13070
13515
|
@classmethod
|
|
@@ -13129,7 +13574,9 @@ def FloorModOptionsEnd(builder):
|
|
|
13129
13574
|
class FloorModOptionsT(object):
|
|
13130
13575
|
|
|
13131
13576
|
# FloorModOptionsT
|
|
13132
|
-
def __init__(
|
|
13577
|
+
def __init__(
|
|
13578
|
+
self,
|
|
13579
|
+
):
|
|
13133
13580
|
pass
|
|
13134
13581
|
|
|
13135
13582
|
@classmethod
|
|
@@ -13194,7 +13641,9 @@ def RangeOptionsEnd(builder):
|
|
|
13194
13641
|
class RangeOptionsT(object):
|
|
13195
13642
|
|
|
13196
13643
|
# RangeOptionsT
|
|
13197
|
-
def __init__(
|
|
13644
|
+
def __init__(
|
|
13645
|
+
self,
|
|
13646
|
+
):
|
|
13198
13647
|
pass
|
|
13199
13648
|
|
|
13200
13649
|
@classmethod
|
|
@@ -13269,8 +13718,11 @@ def LeakyReluOptionsEnd(builder):
|
|
|
13269
13718
|
class LeakyReluOptionsT(object):
|
|
13270
13719
|
|
|
13271
13720
|
# LeakyReluOptionsT
|
|
13272
|
-
def __init__(
|
|
13273
|
-
self
|
|
13721
|
+
def __init__(
|
|
13722
|
+
self,
|
|
13723
|
+
alpha = 0.0,
|
|
13724
|
+
):
|
|
13725
|
+
self.alpha = alpha # type: float
|
|
13274
13726
|
|
|
13275
13727
|
@classmethod
|
|
13276
13728
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -13336,7 +13788,9 @@ def SquaredDifferenceOptionsEnd(builder):
|
|
|
13336
13788
|
class SquaredDifferenceOptionsT(object):
|
|
13337
13789
|
|
|
13338
13790
|
# SquaredDifferenceOptionsT
|
|
13339
|
-
def __init__(
|
|
13791
|
+
def __init__(
|
|
13792
|
+
self,
|
|
13793
|
+
):
|
|
13340
13794
|
pass
|
|
13341
13795
|
|
|
13342
13796
|
@classmethod
|
|
@@ -13411,8 +13865,11 @@ def MirrorPadOptionsEnd(builder):
|
|
|
13411
13865
|
class MirrorPadOptionsT(object):
|
|
13412
13866
|
|
|
13413
13867
|
# MirrorPadOptionsT
|
|
13414
|
-
def __init__(
|
|
13415
|
-
self
|
|
13868
|
+
def __init__(
|
|
13869
|
+
self,
|
|
13870
|
+
mode = 0,
|
|
13871
|
+
):
|
|
13872
|
+
self.mode = mode # type: int
|
|
13416
13873
|
|
|
13417
13874
|
@classmethod
|
|
13418
13875
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -13488,8 +13945,11 @@ def UniqueOptionsEnd(builder):
|
|
|
13488
13945
|
class UniqueOptionsT(object):
|
|
13489
13946
|
|
|
13490
13947
|
# UniqueOptionsT
|
|
13491
|
-
def __init__(
|
|
13492
|
-
self
|
|
13948
|
+
def __init__(
|
|
13949
|
+
self,
|
|
13950
|
+
idxOutType = 2,
|
|
13951
|
+
):
|
|
13952
|
+
self.idxOutType = idxOutType # type: int
|
|
13493
13953
|
|
|
13494
13954
|
@classmethod
|
|
13495
13955
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -13555,7 +14015,9 @@ def ReverseV2OptionsEnd(builder):
|
|
|
13555
14015
|
class ReverseV2OptionsT(object):
|
|
13556
14016
|
|
|
13557
14017
|
# ReverseV2OptionsT
|
|
13558
|
-
def __init__(
|
|
14018
|
+
def __init__(
|
|
14019
|
+
self,
|
|
14020
|
+
):
|
|
13559
14021
|
pass
|
|
13560
14022
|
|
|
13561
14023
|
@classmethod
|
|
@@ -13620,7 +14082,9 @@ def AddNOptionsEnd(builder):
|
|
|
13620
14082
|
class AddNOptionsT(object):
|
|
13621
14083
|
|
|
13622
14084
|
# AddNOptionsT
|
|
13623
|
-
def __init__(
|
|
14085
|
+
def __init__(
|
|
14086
|
+
self,
|
|
14087
|
+
):
|
|
13624
14088
|
pass
|
|
13625
14089
|
|
|
13626
14090
|
@classmethod
|
|
@@ -13685,7 +14149,9 @@ def GatherNdOptionsEnd(builder):
|
|
|
13685
14149
|
class GatherNdOptionsT(object):
|
|
13686
14150
|
|
|
13687
14151
|
# GatherNdOptionsT
|
|
13688
|
-
def __init__(
|
|
14152
|
+
def __init__(
|
|
14153
|
+
self,
|
|
14154
|
+
):
|
|
13689
14155
|
pass
|
|
13690
14156
|
|
|
13691
14157
|
@classmethod
|
|
@@ -13750,7 +14216,9 @@ def WhereOptionsEnd(builder):
|
|
|
13750
14216
|
class WhereOptionsT(object):
|
|
13751
14217
|
|
|
13752
14218
|
# WhereOptionsT
|
|
13753
|
-
def __init__(
|
|
14219
|
+
def __init__(
|
|
14220
|
+
self,
|
|
14221
|
+
):
|
|
13754
14222
|
pass
|
|
13755
14223
|
|
|
13756
14224
|
@classmethod
|
|
@@ -13835,9 +14303,13 @@ def ReverseSequenceOptionsEnd(builder):
|
|
|
13835
14303
|
class ReverseSequenceOptionsT(object):
|
|
13836
14304
|
|
|
13837
14305
|
# ReverseSequenceOptionsT
|
|
13838
|
-
def __init__(
|
|
13839
|
-
self
|
|
13840
|
-
|
|
14306
|
+
def __init__(
|
|
14307
|
+
self,
|
|
14308
|
+
seqDim = 0,
|
|
14309
|
+
batchDim = 0,
|
|
14310
|
+
):
|
|
14311
|
+
self.seqDim = seqDim # type: int
|
|
14312
|
+
self.batchDim = batchDim # type: int
|
|
13841
14313
|
|
|
13842
14314
|
@classmethod
|
|
13843
14315
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -13905,7 +14377,9 @@ def MatrixDiagOptionsEnd(builder):
|
|
|
13905
14377
|
class MatrixDiagOptionsT(object):
|
|
13906
14378
|
|
|
13907
14379
|
# MatrixDiagOptionsT
|
|
13908
|
-
def __init__(
|
|
14380
|
+
def __init__(
|
|
14381
|
+
self,
|
|
14382
|
+
):
|
|
13909
14383
|
pass
|
|
13910
14384
|
|
|
13911
14385
|
@classmethod
|
|
@@ -13970,7 +14444,9 @@ def QuantizeOptionsEnd(builder):
|
|
|
13970
14444
|
class QuantizeOptionsT(object):
|
|
13971
14445
|
|
|
13972
14446
|
# QuantizeOptionsT
|
|
13973
|
-
def __init__(
|
|
14447
|
+
def __init__(
|
|
14448
|
+
self,
|
|
14449
|
+
):
|
|
13974
14450
|
pass
|
|
13975
14451
|
|
|
13976
14452
|
@classmethod
|
|
@@ -14035,7 +14511,9 @@ def MatrixSetDiagOptionsEnd(builder):
|
|
|
14035
14511
|
class MatrixSetDiagOptionsT(object):
|
|
14036
14512
|
|
|
14037
14513
|
# MatrixSetDiagOptionsT
|
|
14038
|
-
def __init__(
|
|
14514
|
+
def __init__(
|
|
14515
|
+
self,
|
|
14516
|
+
):
|
|
14039
14517
|
pass
|
|
14040
14518
|
|
|
14041
14519
|
@classmethod
|
|
@@ -14120,9 +14598,13 @@ def IfOptionsEnd(builder):
|
|
|
14120
14598
|
class IfOptionsT(object):
|
|
14121
14599
|
|
|
14122
14600
|
# IfOptionsT
|
|
14123
|
-
def __init__(
|
|
14124
|
-
self
|
|
14125
|
-
|
|
14601
|
+
def __init__(
|
|
14602
|
+
self,
|
|
14603
|
+
thenSubgraphIndex = 0,
|
|
14604
|
+
elseSubgraphIndex = 0,
|
|
14605
|
+
):
|
|
14606
|
+
self.thenSubgraphIndex = thenSubgraphIndex # type: int
|
|
14607
|
+
self.elseSubgraphIndex = elseSubgraphIndex # type: int
|
|
14126
14608
|
|
|
14127
14609
|
@classmethod
|
|
14128
14610
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -14200,8 +14682,11 @@ def CallOnceOptionsEnd(builder):
|
|
|
14200
14682
|
class CallOnceOptionsT(object):
|
|
14201
14683
|
|
|
14202
14684
|
# CallOnceOptionsT
|
|
14203
|
-
def __init__(
|
|
14204
|
-
self
|
|
14685
|
+
def __init__(
|
|
14686
|
+
self,
|
|
14687
|
+
initSubgraphIndex = 0,
|
|
14688
|
+
):
|
|
14689
|
+
self.initSubgraphIndex = initSubgraphIndex # type: int
|
|
14205
14690
|
|
|
14206
14691
|
@classmethod
|
|
14207
14692
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -14287,9 +14772,13 @@ def WhileOptionsEnd(builder):
|
|
|
14287
14772
|
class WhileOptionsT(object):
|
|
14288
14773
|
|
|
14289
14774
|
# WhileOptionsT
|
|
14290
|
-
def __init__(
|
|
14291
|
-
self
|
|
14292
|
-
|
|
14775
|
+
def __init__(
|
|
14776
|
+
self,
|
|
14777
|
+
condSubgraphIndex = 0,
|
|
14778
|
+
bodySubgraphIndex = 0,
|
|
14779
|
+
):
|
|
14780
|
+
self.condSubgraphIndex = condSubgraphIndex # type: int
|
|
14781
|
+
self.bodySubgraphIndex = bodySubgraphIndex # type: int
|
|
14293
14782
|
|
|
14294
14783
|
@classmethod
|
|
14295
14784
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -14357,7 +14846,9 @@ def NonMaxSuppressionV4OptionsEnd(builder):
|
|
|
14357
14846
|
class NonMaxSuppressionV4OptionsT(object):
|
|
14358
14847
|
|
|
14359
14848
|
# NonMaxSuppressionV4OptionsT
|
|
14360
|
-
def __init__(
|
|
14849
|
+
def __init__(
|
|
14850
|
+
self,
|
|
14851
|
+
):
|
|
14361
14852
|
pass
|
|
14362
14853
|
|
|
14363
14854
|
@classmethod
|
|
@@ -14422,7 +14913,9 @@ def NonMaxSuppressionV5OptionsEnd(builder):
|
|
|
14422
14913
|
class NonMaxSuppressionV5OptionsT(object):
|
|
14423
14914
|
|
|
14424
14915
|
# NonMaxSuppressionV5OptionsT
|
|
14425
|
-
def __init__(
|
|
14916
|
+
def __init__(
|
|
14917
|
+
self,
|
|
14918
|
+
):
|
|
14426
14919
|
pass
|
|
14427
14920
|
|
|
14428
14921
|
@classmethod
|
|
@@ -14487,7 +14980,9 @@ def ScatterNdOptionsEnd(builder):
|
|
|
14487
14980
|
class ScatterNdOptionsT(object):
|
|
14488
14981
|
|
|
14489
14982
|
# ScatterNdOptionsT
|
|
14490
|
-
def __init__(
|
|
14983
|
+
def __init__(
|
|
14984
|
+
self,
|
|
14985
|
+
):
|
|
14491
14986
|
pass
|
|
14492
14987
|
|
|
14493
14988
|
@classmethod
|
|
@@ -14552,7 +15047,9 @@ def SelectV2OptionsEnd(builder):
|
|
|
14552
15047
|
class SelectV2OptionsT(object):
|
|
14553
15048
|
|
|
14554
15049
|
# SelectV2OptionsT
|
|
14555
|
-
def __init__(
|
|
15050
|
+
def __init__(
|
|
15051
|
+
self,
|
|
15052
|
+
):
|
|
14556
15053
|
pass
|
|
14557
15054
|
|
|
14558
15055
|
@classmethod
|
|
@@ -14617,7 +15114,9 @@ def DensifyOptionsEnd(builder):
|
|
|
14617
15114
|
class DensifyOptionsT(object):
|
|
14618
15115
|
|
|
14619
15116
|
# DensifyOptionsT
|
|
14620
|
-
def __init__(
|
|
15117
|
+
def __init__(
|
|
15118
|
+
self,
|
|
15119
|
+
):
|
|
14621
15120
|
pass
|
|
14622
15121
|
|
|
14623
15122
|
@classmethod
|
|
@@ -14682,7 +15181,9 @@ def SegmentSumOptionsEnd(builder):
|
|
|
14682
15181
|
class SegmentSumOptionsT(object):
|
|
14683
15182
|
|
|
14684
15183
|
# SegmentSumOptionsT
|
|
14685
|
-
def __init__(
|
|
15184
|
+
def __init__(
|
|
15185
|
+
self,
|
|
15186
|
+
):
|
|
14686
15187
|
pass
|
|
14687
15188
|
|
|
14688
15189
|
@classmethod
|
|
@@ -14777,10 +15278,15 @@ def BatchMatMulOptionsEnd(builder):
|
|
|
14777
15278
|
class BatchMatMulOptionsT(object):
|
|
14778
15279
|
|
|
14779
15280
|
# BatchMatMulOptionsT
|
|
14780
|
-
def __init__(
|
|
14781
|
-
self
|
|
14782
|
-
|
|
14783
|
-
|
|
15281
|
+
def __init__(
|
|
15282
|
+
self,
|
|
15283
|
+
adjX = False,
|
|
15284
|
+
adjY = False,
|
|
15285
|
+
asymmetricQuantizeInputs = False,
|
|
15286
|
+
):
|
|
15287
|
+
self.adjX = adjX # type: bool
|
|
15288
|
+
self.adjY = adjY # type: bool
|
|
15289
|
+
self.asymmetricQuantizeInputs = asymmetricQuantizeInputs # type: bool
|
|
14784
15290
|
|
|
14785
15291
|
@classmethod
|
|
14786
15292
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -14870,9 +15376,13 @@ def CumsumOptionsEnd(builder):
|
|
|
14870
15376
|
class CumsumOptionsT(object):
|
|
14871
15377
|
|
|
14872
15378
|
# CumsumOptionsT
|
|
14873
|
-
def __init__(
|
|
14874
|
-
self
|
|
14875
|
-
|
|
15379
|
+
def __init__(
|
|
15380
|
+
self,
|
|
15381
|
+
exclusive = False,
|
|
15382
|
+
reverse = False,
|
|
15383
|
+
):
|
|
15384
|
+
self.exclusive = exclusive # type: bool
|
|
15385
|
+
self.reverse = reverse # type: bool
|
|
14876
15386
|
|
|
14877
15387
|
@classmethod
|
|
14878
15388
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -14940,7 +15450,9 @@ def BroadcastToOptionsEnd(builder):
|
|
|
14940
15450
|
class BroadcastToOptionsT(object):
|
|
14941
15451
|
|
|
14942
15452
|
# BroadcastToOptionsT
|
|
14943
|
-
def __init__(
|
|
15453
|
+
def __init__(
|
|
15454
|
+
self,
|
|
15455
|
+
):
|
|
14944
15456
|
pass
|
|
14945
15457
|
|
|
14946
15458
|
@classmethod
|
|
@@ -15005,7 +15517,9 @@ def Rfft2dOptionsEnd(builder):
|
|
|
15005
15517
|
class Rfft2dOptionsT(object):
|
|
15006
15518
|
|
|
15007
15519
|
# Rfft2dOptionsT
|
|
15008
|
-
def __init__(
|
|
15520
|
+
def __init__(
|
|
15521
|
+
self,
|
|
15522
|
+
):
|
|
15009
15523
|
pass
|
|
15010
15524
|
|
|
15011
15525
|
@classmethod
|
|
@@ -15100,10 +15614,15 @@ def HashtableOptionsEnd(builder):
|
|
|
15100
15614
|
class HashtableOptionsT(object):
|
|
15101
15615
|
|
|
15102
15616
|
# HashtableOptionsT
|
|
15103
|
-
def __init__(
|
|
15104
|
-
self
|
|
15105
|
-
|
|
15106
|
-
|
|
15617
|
+
def __init__(
|
|
15618
|
+
self,
|
|
15619
|
+
tableId = 0,
|
|
15620
|
+
keyDtype = 0,
|
|
15621
|
+
valueDtype = 0,
|
|
15622
|
+
):
|
|
15623
|
+
self.tableId = tableId # type: int
|
|
15624
|
+
self.keyDtype = keyDtype # type: int
|
|
15625
|
+
self.valueDtype = valueDtype # type: int
|
|
15107
15626
|
|
|
15108
15627
|
@classmethod
|
|
15109
15628
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -15173,7 +15692,9 @@ def HashtableFindOptionsEnd(builder):
|
|
|
15173
15692
|
class HashtableFindOptionsT(object):
|
|
15174
15693
|
|
|
15175
15694
|
# HashtableFindOptionsT
|
|
15176
|
-
def __init__(
|
|
15695
|
+
def __init__(
|
|
15696
|
+
self,
|
|
15697
|
+
):
|
|
15177
15698
|
pass
|
|
15178
15699
|
|
|
15179
15700
|
@classmethod
|
|
@@ -15238,7 +15759,9 @@ def HashtableImportOptionsEnd(builder):
|
|
|
15238
15759
|
class HashtableImportOptionsT(object):
|
|
15239
15760
|
|
|
15240
15761
|
# HashtableImportOptionsT
|
|
15241
|
-
def __init__(
|
|
15762
|
+
def __init__(
|
|
15763
|
+
self,
|
|
15764
|
+
):
|
|
15242
15765
|
pass
|
|
15243
15766
|
|
|
15244
15767
|
@classmethod
|
|
@@ -15303,7 +15826,9 @@ def HashtableSizeOptionsEnd(builder):
|
|
|
15303
15826
|
class HashtableSizeOptionsT(object):
|
|
15304
15827
|
|
|
15305
15828
|
# HashtableSizeOptionsT
|
|
15306
|
-
def __init__(
|
|
15829
|
+
def __init__(
|
|
15830
|
+
self,
|
|
15831
|
+
):
|
|
15307
15832
|
pass
|
|
15308
15833
|
|
|
15309
15834
|
@classmethod
|
|
@@ -15388,9 +15913,13 @@ def VarHandleOptionsEnd(builder):
|
|
|
15388
15913
|
class VarHandleOptionsT(object):
|
|
15389
15914
|
|
|
15390
15915
|
# VarHandleOptionsT
|
|
15391
|
-
def __init__(
|
|
15392
|
-
self
|
|
15393
|
-
|
|
15916
|
+
def __init__(
|
|
15917
|
+
self,
|
|
15918
|
+
container = None,
|
|
15919
|
+
sharedName = None,
|
|
15920
|
+
):
|
|
15921
|
+
self.container = container # type: Optional[str]
|
|
15922
|
+
self.sharedName = sharedName # type: Optional[str]
|
|
15394
15923
|
|
|
15395
15924
|
@classmethod
|
|
15396
15925
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -15464,7 +15993,9 @@ def ReadVariableOptionsEnd(builder):
|
|
|
15464
15993
|
class ReadVariableOptionsT(object):
|
|
15465
15994
|
|
|
15466
15995
|
# ReadVariableOptionsT
|
|
15467
|
-
def __init__(
|
|
15996
|
+
def __init__(
|
|
15997
|
+
self,
|
|
15998
|
+
):
|
|
15468
15999
|
pass
|
|
15469
16000
|
|
|
15470
16001
|
@classmethod
|
|
@@ -15529,7 +16060,9 @@ def AssignVariableOptionsEnd(builder):
|
|
|
15529
16060
|
class AssignVariableOptionsT(object):
|
|
15530
16061
|
|
|
15531
16062
|
# AssignVariableOptionsT
|
|
15532
|
-
def __init__(
|
|
16063
|
+
def __init__(
|
|
16064
|
+
self,
|
|
16065
|
+
):
|
|
15533
16066
|
pass
|
|
15534
16067
|
|
|
15535
16068
|
@classmethod
|
|
@@ -15614,9 +16147,13 @@ def RandomOptionsEnd(builder):
|
|
|
15614
16147
|
class RandomOptionsT(object):
|
|
15615
16148
|
|
|
15616
16149
|
# RandomOptionsT
|
|
15617
|
-
def __init__(
|
|
15618
|
-
self
|
|
15619
|
-
|
|
16150
|
+
def __init__(
|
|
16151
|
+
self,
|
|
16152
|
+
seed = 0,
|
|
16153
|
+
seed2 = 0,
|
|
16154
|
+
):
|
|
16155
|
+
self.seed = seed # type: int
|
|
16156
|
+
self.seed2 = seed2 # type: int
|
|
15620
16157
|
|
|
15621
16158
|
@classmethod
|
|
15622
16159
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -15721,8 +16258,11 @@ except:
|
|
|
15721
16258
|
class BucketizeOptionsT(object):
|
|
15722
16259
|
|
|
15723
16260
|
# BucketizeOptionsT
|
|
15724
|
-
def __init__(
|
|
15725
|
-
self
|
|
16261
|
+
def __init__(
|
|
16262
|
+
self,
|
|
16263
|
+
boundaries = None,
|
|
16264
|
+
):
|
|
16265
|
+
self.boundaries = boundaries # type: Optional[List[float]]
|
|
15726
16266
|
|
|
15727
16267
|
@classmethod
|
|
15728
16268
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -15813,8 +16353,11 @@ def GeluOptionsEnd(builder):
|
|
|
15813
16353
|
class GeluOptionsT(object):
|
|
15814
16354
|
|
|
15815
16355
|
# GeluOptionsT
|
|
15816
|
-
def __init__(
|
|
15817
|
-
self
|
|
16356
|
+
def __init__(
|
|
16357
|
+
self,
|
|
16358
|
+
approximate = False,
|
|
16359
|
+
):
|
|
16360
|
+
self.approximate = approximate # type: bool
|
|
15818
16361
|
|
|
15819
16362
|
@classmethod
|
|
15820
16363
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -15880,7 +16423,9 @@ def DynamicUpdateSliceOptionsEnd(builder):
|
|
|
15880
16423
|
class DynamicUpdateSliceOptionsT(object):
|
|
15881
16424
|
|
|
15882
16425
|
# DynamicUpdateSliceOptionsT
|
|
15883
|
-
def __init__(
|
|
16426
|
+
def __init__(
|
|
16427
|
+
self,
|
|
16428
|
+
):
|
|
15884
16429
|
pass
|
|
15885
16430
|
|
|
15886
16431
|
@classmethod
|
|
@@ -15945,7 +16490,9 @@ def UnsortedSegmentProdOptionsEnd(builder):
|
|
|
15945
16490
|
class UnsortedSegmentProdOptionsT(object):
|
|
15946
16491
|
|
|
15947
16492
|
# UnsortedSegmentProdOptionsT
|
|
15948
|
-
def __init__(
|
|
16493
|
+
def __init__(
|
|
16494
|
+
self,
|
|
16495
|
+
):
|
|
15949
16496
|
pass
|
|
15950
16497
|
|
|
15951
16498
|
@classmethod
|
|
@@ -16010,7 +16557,9 @@ def UnsortedSegmentMaxOptionsEnd(builder):
|
|
|
16010
16557
|
class UnsortedSegmentMaxOptionsT(object):
|
|
16011
16558
|
|
|
16012
16559
|
# UnsortedSegmentMaxOptionsT
|
|
16013
|
-
def __init__(
|
|
16560
|
+
def __init__(
|
|
16561
|
+
self,
|
|
16562
|
+
):
|
|
16014
16563
|
pass
|
|
16015
16564
|
|
|
16016
16565
|
@classmethod
|
|
@@ -16075,7 +16624,9 @@ def UnsortedSegmentSumOptionsEnd(builder):
|
|
|
16075
16624
|
class UnsortedSegmentSumOptionsT(object):
|
|
16076
16625
|
|
|
16077
16626
|
# UnsortedSegmentSumOptionsT
|
|
16078
|
-
def __init__(
|
|
16627
|
+
def __init__(
|
|
16628
|
+
self,
|
|
16629
|
+
):
|
|
16079
16630
|
pass
|
|
16080
16631
|
|
|
16081
16632
|
@classmethod
|
|
@@ -16140,7 +16691,9 @@ def ATan2OptionsEnd(builder):
|
|
|
16140
16691
|
class ATan2OptionsT(object):
|
|
16141
16692
|
|
|
16142
16693
|
# ATan2OptionsT
|
|
16143
|
-
def __init__(
|
|
16694
|
+
def __init__(
|
|
16695
|
+
self,
|
|
16696
|
+
):
|
|
16144
16697
|
pass
|
|
16145
16698
|
|
|
16146
16699
|
@classmethod
|
|
@@ -16205,7 +16758,9 @@ def UnsortedSegmentMinOptionsEnd(builder):
|
|
|
16205
16758
|
class UnsortedSegmentMinOptionsT(object):
|
|
16206
16759
|
|
|
16207
16760
|
# UnsortedSegmentMinOptionsT
|
|
16208
|
-
def __init__(
|
|
16761
|
+
def __init__(
|
|
16762
|
+
self,
|
|
16763
|
+
):
|
|
16209
16764
|
pass
|
|
16210
16765
|
|
|
16211
16766
|
@classmethod
|
|
@@ -16270,7 +16825,9 @@ def SignOptionsEnd(builder):
|
|
|
16270
16825
|
class SignOptionsT(object):
|
|
16271
16826
|
|
|
16272
16827
|
# SignOptionsT
|
|
16273
|
-
def __init__(
|
|
16828
|
+
def __init__(
|
|
16829
|
+
self,
|
|
16830
|
+
):
|
|
16274
16831
|
pass
|
|
16275
16832
|
|
|
16276
16833
|
@classmethod
|
|
@@ -16335,7 +16892,9 @@ def BitcastOptionsEnd(builder):
|
|
|
16335
16892
|
class BitcastOptionsT(object):
|
|
16336
16893
|
|
|
16337
16894
|
# BitcastOptionsT
|
|
16338
|
-
def __init__(
|
|
16895
|
+
def __init__(
|
|
16896
|
+
self,
|
|
16897
|
+
):
|
|
16339
16898
|
pass
|
|
16340
16899
|
|
|
16341
16900
|
@classmethod
|
|
@@ -16400,7 +16959,9 @@ def BitwiseXorOptionsEnd(builder):
|
|
|
16400
16959
|
class BitwiseXorOptionsT(object):
|
|
16401
16960
|
|
|
16402
16961
|
# BitwiseXorOptionsT
|
|
16403
|
-
def __init__(
|
|
16962
|
+
def __init__(
|
|
16963
|
+
self,
|
|
16964
|
+
):
|
|
16404
16965
|
pass
|
|
16405
16966
|
|
|
16406
16967
|
@classmethod
|
|
@@ -16465,7 +17026,9 @@ def RightShiftOptionsEnd(builder):
|
|
|
16465
17026
|
class RightShiftOptionsT(object):
|
|
16466
17027
|
|
|
16467
17028
|
# RightShiftOptionsT
|
|
16468
|
-
def __init__(
|
|
17029
|
+
def __init__(
|
|
17030
|
+
self,
|
|
17031
|
+
):
|
|
16469
17032
|
pass
|
|
16470
17033
|
|
|
16471
17034
|
@classmethod
|
|
@@ -16530,7 +17093,9 @@ def DilateOptionsEnd(builder):
|
|
|
16530
17093
|
class DilateOptionsT(object):
|
|
16531
17094
|
|
|
16532
17095
|
# DilateOptionsT
|
|
16533
|
-
def __init__(
|
|
17096
|
+
def __init__(
|
|
17097
|
+
self,
|
|
17098
|
+
):
|
|
16534
17099
|
pass
|
|
16535
17100
|
|
|
16536
17101
|
@classmethod
|
|
@@ -16605,8 +17170,11 @@ def ReduceWindowOptionsEnd(builder):
|
|
|
16605
17170
|
class ReduceWindowOptionsT(object):
|
|
16606
17171
|
|
|
16607
17172
|
# ReduceWindowOptionsT
|
|
16608
|
-
def __init__(
|
|
16609
|
-
self
|
|
17173
|
+
def __init__(
|
|
17174
|
+
self,
|
|
17175
|
+
reduceFunction = 0,
|
|
17176
|
+
):
|
|
17177
|
+
self.reduceFunction = reduceFunction # type: int
|
|
16610
17178
|
|
|
16611
17179
|
@classmethod
|
|
16612
17180
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -16712,11 +17280,17 @@ def OperatorCodeEnd(builder):
|
|
|
16712
17280
|
class OperatorCodeT(object):
|
|
16713
17281
|
|
|
16714
17282
|
# OperatorCodeT
|
|
16715
|
-
def __init__(
|
|
16716
|
-
self
|
|
16717
|
-
|
|
16718
|
-
|
|
16719
|
-
|
|
17283
|
+
def __init__(
|
|
17284
|
+
self,
|
|
17285
|
+
deprecatedBuiltinCode = 0,
|
|
17286
|
+
customCode = None,
|
|
17287
|
+
version = 1,
|
|
17288
|
+
builtinCode = 0,
|
|
17289
|
+
):
|
|
17290
|
+
self.deprecatedBuiltinCode = deprecatedBuiltinCode # type: int
|
|
17291
|
+
self.customCode = customCode # type: Optional[str]
|
|
17292
|
+
self.version = version # type: int
|
|
17293
|
+
self.builtinCode = builtinCode # type: int
|
|
16720
17294
|
|
|
16721
17295
|
@classmethod
|
|
16722
17296
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -16868,12 +17442,19 @@ except:
|
|
|
16868
17442
|
class StableHLOCompositeOptionsT(object):
|
|
16869
17443
|
|
|
16870
17444
|
# StableHLOCompositeOptionsT
|
|
16871
|
-
def __init__(
|
|
16872
|
-
self
|
|
16873
|
-
|
|
16874
|
-
|
|
16875
|
-
|
|
16876
|
-
|
|
17445
|
+
def __init__(
|
|
17446
|
+
self,
|
|
17447
|
+
name = None,
|
|
17448
|
+
decompositionSubgraphIndex = 0,
|
|
17449
|
+
compositeAttributes = None,
|
|
17450
|
+
compositeAttributesFormat = 0,
|
|
17451
|
+
version = 0,
|
|
17452
|
+
):
|
|
17453
|
+
self.name = name # type: Optional[str]
|
|
17454
|
+
self.decompositionSubgraphIndex = decompositionSubgraphIndex # type: int
|
|
17455
|
+
self.compositeAttributes = compositeAttributes # type: Optional[List[int]]
|
|
17456
|
+
self.compositeAttributesFormat = compositeAttributesFormat # type: int
|
|
17457
|
+
self.version = version # type: int
|
|
16877
17458
|
|
|
16878
17459
|
@classmethod
|
|
16879
17460
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -16965,7 +17546,9 @@ def StablehloShiftLeftOptionsEnd(builder):
|
|
|
16965
17546
|
class StablehloShiftLeftOptionsT(object):
|
|
16966
17547
|
|
|
16967
17548
|
# StablehloShiftLeftOptionsT
|
|
16968
|
-
def __init__(
|
|
17549
|
+
def __init__(
|
|
17550
|
+
self,
|
|
17551
|
+
):
|
|
16969
17552
|
pass
|
|
16970
17553
|
|
|
16971
17554
|
@classmethod
|
|
@@ -17295,21 +17878,37 @@ except:
|
|
|
17295
17878
|
class OperatorT(object):
|
|
17296
17879
|
|
|
17297
17880
|
# OperatorT
|
|
17298
|
-
def __init__(
|
|
17299
|
-
self
|
|
17300
|
-
|
|
17301
|
-
|
|
17302
|
-
|
|
17303
|
-
|
|
17304
|
-
|
|
17305
|
-
|
|
17306
|
-
|
|
17307
|
-
|
|
17308
|
-
|
|
17309
|
-
|
|
17310
|
-
|
|
17311
|
-
|
|
17312
|
-
|
|
17881
|
+
def __init__(
|
|
17882
|
+
self,
|
|
17883
|
+
opcodeIndex = 0,
|
|
17884
|
+
inputs = None,
|
|
17885
|
+
outputs = None,
|
|
17886
|
+
builtinOptionsType = 0,
|
|
17887
|
+
builtinOptions = None,
|
|
17888
|
+
customOptions = None,
|
|
17889
|
+
customOptionsFormat = 0,
|
|
17890
|
+
mutatingVariableInputs = None,
|
|
17891
|
+
intermediates = None,
|
|
17892
|
+
largeCustomOptionsOffset = 0,
|
|
17893
|
+
largeCustomOptionsSize = 0,
|
|
17894
|
+
builtinOptions2Type = 0,
|
|
17895
|
+
builtinOptions2 = None,
|
|
17896
|
+
debugMetadataIndex = -1,
|
|
17897
|
+
):
|
|
17898
|
+
self.opcodeIndex = opcodeIndex # type: int
|
|
17899
|
+
self.inputs = inputs # type: Optional[List[int]]
|
|
17900
|
+
self.outputs = outputs # type: Optional[List[int]]
|
|
17901
|
+
self.builtinOptionsType = builtinOptionsType # type: int
|
|
17902
|
+
self.builtinOptions = builtinOptions # 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']
|
|
17903
|
+
self.customOptions = customOptions # type: Optional[List[int]]
|
|
17904
|
+
self.customOptionsFormat = customOptionsFormat # type: int
|
|
17905
|
+
self.mutatingVariableInputs = mutatingVariableInputs # type: Optional[List[bool]]
|
|
17906
|
+
self.intermediates = intermediates # type: Optional[List[int]]
|
|
17907
|
+
self.largeCustomOptionsOffset = largeCustomOptionsOffset # type: int
|
|
17908
|
+
self.largeCustomOptionsSize = largeCustomOptionsSize # type: int
|
|
17909
|
+
self.builtinOptions2Type = builtinOptions2Type # type: int
|
|
17910
|
+
self.builtinOptions2 = builtinOptions2 # type: Union[None, 'StablehloConcatenateOptionsT', 'StablehloBroadcastInDimOptionsT', 'StablehloSliceOptionsT', 'StablehloConvolutionOptionsT', 'StablehloCustomCallOptionsT', 'StablehloReduceOptionsT', 'StablehloScatterOptionsT', 'StablehloCompareOptionsT', 'StablehloDynamicSliceOptionsT', 'StablehloPadOptionsT', 'StablehloIotaOptionsT', 'StablehloDotGeneralOptionsT', 'StablehloReduceWindowOptionsT', 'StablehloSortOptionsT', 'StablehloWhileOptionsT', 'StablehloGatherOptionsT', 'StablehloTransposeOptionsT', 'DilateOptionsT', 'StablehloRngBitGeneratorOptionsT', 'ReduceWindowOptionsT', 'StableHLOCompositeOptionsT', 'StablehloShiftLeftOptionsT', 'StablehloCaseOptionsT']
|
|
17911
|
+
self.debugMetadataIndex = debugMetadataIndex # type: int
|
|
17313
17912
|
|
|
17314
17913
|
@classmethod
|
|
17315
17914
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -17632,13 +18231,21 @@ except:
|
|
|
17632
18231
|
class SubGraphT(object):
|
|
17633
18232
|
|
|
17634
18233
|
# SubGraphT
|
|
17635
|
-
def __init__(
|
|
17636
|
-
self
|
|
17637
|
-
|
|
17638
|
-
|
|
17639
|
-
|
|
17640
|
-
|
|
17641
|
-
|
|
18234
|
+
def __init__(
|
|
18235
|
+
self,
|
|
18236
|
+
tensors = None,
|
|
18237
|
+
inputs = None,
|
|
18238
|
+
outputs = None,
|
|
18239
|
+
operators = None,
|
|
18240
|
+
name = None,
|
|
18241
|
+
debugMetadataIndex = -1,
|
|
18242
|
+
):
|
|
18243
|
+
self.tensors = tensors # type: Optional[List[TensorT]]
|
|
18244
|
+
self.inputs = inputs # type: Optional[List[int]]
|
|
18245
|
+
self.outputs = outputs # type: Optional[List[int]]
|
|
18246
|
+
self.operators = operators # type: Optional[List[OperatorT]]
|
|
18247
|
+
self.name = name # type: Optional[str]
|
|
18248
|
+
self.debugMetadataIndex = debugMetadataIndex # type: int
|
|
17642
18249
|
|
|
17643
18250
|
@classmethod
|
|
17644
18251
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -17836,10 +18443,15 @@ except:
|
|
|
17836
18443
|
class BufferT(object):
|
|
17837
18444
|
|
|
17838
18445
|
# BufferT
|
|
17839
|
-
def __init__(
|
|
17840
|
-
self
|
|
17841
|
-
|
|
17842
|
-
|
|
18446
|
+
def __init__(
|
|
18447
|
+
self,
|
|
18448
|
+
data = None,
|
|
18449
|
+
offset = 0,
|
|
18450
|
+
size = 0,
|
|
18451
|
+
):
|
|
18452
|
+
self.data = data # type: Optional[List[int]]
|
|
18453
|
+
self.offset = offset # type: int
|
|
18454
|
+
self.size = size # type: int
|
|
17843
18455
|
|
|
17844
18456
|
@classmethod
|
|
17845
18457
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -17944,9 +18556,13 @@ def MetadataEnd(builder):
|
|
|
17944
18556
|
class MetadataT(object):
|
|
17945
18557
|
|
|
17946
18558
|
# MetadataT
|
|
17947
|
-
def __init__(
|
|
17948
|
-
self
|
|
17949
|
-
|
|
18559
|
+
def __init__(
|
|
18560
|
+
self,
|
|
18561
|
+
name = None,
|
|
18562
|
+
buffer = 0,
|
|
18563
|
+
):
|
|
18564
|
+
self.name = name # type: Optional[str]
|
|
18565
|
+
self.buffer = buffer # type: int
|
|
17950
18566
|
|
|
17951
18567
|
@classmethod
|
|
17952
18568
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -18037,9 +18653,13 @@ def TensorMapEnd(builder):
|
|
|
18037
18653
|
class TensorMapT(object):
|
|
18038
18654
|
|
|
18039
18655
|
# TensorMapT
|
|
18040
|
-
def __init__(
|
|
18041
|
-
self
|
|
18042
|
-
|
|
18656
|
+
def __init__(
|
|
18657
|
+
self,
|
|
18658
|
+
name = None,
|
|
18659
|
+
tensorIndex = 0,
|
|
18660
|
+
):
|
|
18661
|
+
self.name = name # type: Optional[str]
|
|
18662
|
+
self.tensorIndex = tensorIndex # type: int
|
|
18043
18663
|
|
|
18044
18664
|
@classmethod
|
|
18045
18665
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -18194,11 +18814,17 @@ except:
|
|
|
18194
18814
|
class SignatureDefT(object):
|
|
18195
18815
|
|
|
18196
18816
|
# SignatureDefT
|
|
18197
|
-
def __init__(
|
|
18198
|
-
self
|
|
18199
|
-
|
|
18200
|
-
|
|
18201
|
-
|
|
18817
|
+
def __init__(
|
|
18818
|
+
self,
|
|
18819
|
+
inputs = None,
|
|
18820
|
+
outputs = None,
|
|
18821
|
+
signatureKey = None,
|
|
18822
|
+
subgraphIndex = 0,
|
|
18823
|
+
):
|
|
18824
|
+
self.inputs = inputs # type: Optional[List[TensorMapT]]
|
|
18825
|
+
self.outputs = outputs # type: Optional[List[TensorMapT]]
|
|
18826
|
+
self.signatureKey = signatureKey # type: Optional[str]
|
|
18827
|
+
self.subgraphIndex = subgraphIndex # type: int
|
|
18202
18828
|
|
|
18203
18829
|
@classmethod
|
|
18204
18830
|
def InitFromBuf(cls, buf, pos):
|
|
@@ -18512,15 +19138,25 @@ except:
|
|
|
18512
19138
|
class ModelT(object):
|
|
18513
19139
|
|
|
18514
19140
|
# ModelT
|
|
18515
|
-
def __init__(
|
|
18516
|
-
self
|
|
18517
|
-
|
|
18518
|
-
|
|
18519
|
-
|
|
18520
|
-
|
|
18521
|
-
|
|
18522
|
-
|
|
18523
|
-
|
|
19141
|
+
def __init__(
|
|
19142
|
+
self,
|
|
19143
|
+
version = 0,
|
|
19144
|
+
operatorCodes = None,
|
|
19145
|
+
subgraphs = None,
|
|
19146
|
+
description = None,
|
|
19147
|
+
buffers = None,
|
|
19148
|
+
metadataBuffer = None,
|
|
19149
|
+
metadata = None,
|
|
19150
|
+
signatureDefs = None,
|
|
19151
|
+
):
|
|
19152
|
+
self.version = version # type: int
|
|
19153
|
+
self.operatorCodes = operatorCodes # type: Optional[List[OperatorCodeT]]
|
|
19154
|
+
self.subgraphs = subgraphs # type: Optional[List[SubGraphT]]
|
|
19155
|
+
self.description = description # type: Optional[str]
|
|
19156
|
+
self.buffers = buffers # type: Optional[List[BufferT]]
|
|
19157
|
+
self.metadataBuffer = metadataBuffer # type: Optional[List[int]]
|
|
19158
|
+
self.metadata = metadata # type: Optional[List[MetadataT]]
|
|
19159
|
+
self.signatureDefs = signatureDefs # type: Optional[List[SignatureDefT]]
|
|
18524
19160
|
|
|
18525
19161
|
@classmethod
|
|
18526
19162
|
def InitFromBuf(cls, buf, pos):
|