@@ -1592,6 +1592,7 @@ class BuiltinOperator(object):
1592
1592
STABLEHLO_WHILE = 200
1593
1593
STABLEHLO_GATHER = 201
1594
1594
STABLEHLO_TRANSPOSE = 202
1595
+ DILATE = 203
1595
1596
# automatically generated by the FlatBuffers compiler, do not modify
1596
1597
1597
1598
# namespace: tflite
@@ -2005,6 +2006,7 @@ class BuiltinOptions2(object):
2005
2006
StablehloWhileOptions = 15
2006
2007
StablehloGatherOptions = 16
2007
2008
StablehloTransposeOptions = 17
2009
+ DilateOptions = 18
2008
2010
2009
2011
def BuiltinOptions2Creator(unionType, table):
2010
2012
from flatbuffers.table import Table
@@ -2044,6 +2046,8 @@ def BuiltinOptions2Creator(unionType, table):
2044
2046
return StablehloGatherOptionsT.InitFromBuf(table.Bytes, table.Pos)
2045
2047
if unionType == BuiltinOptions2().StablehloTransposeOptions:
2046
2048
return StablehloTransposeOptionsT.InitFromBuf(table.Bytes, table.Pos)
2049
+ if unionType == BuiltinOptions2().DilateOptions:
2050
+ return DilateOptionsT.InitFromBuf(table.Bytes, table.Pos)
2047
2051
return None
2048
2052
# automatically generated by the FlatBuffers compiler, do not modify
2049
2053
@@ -3515,6 +3519,70 @@ def Pack(self, builder):
3515
3519
from flatbuffers.compat import import_numpy
3516
3520
np = import_numpy()
3517
3521
3522
+ class DilateOptions(object):
3523
+ __slots__ = ['_tab']
3524
+
3525
+ @classmethod
3526
+ def GetRootAs(cls, buf, offset=0):
3527
+ n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
3528
+ x = DilateOptions()
3529
+ x.Init(buf, n + offset)
3530
+ return x
3531
+
3532
+ @classmethod
3533
+ def GetRootAsDilateOptions(cls, buf, offset=0):
3534
+ """This method is deprecated. Please switch to GetRootAs."""
3535
+ return cls.GetRootAs(buf, offset)
3536
+ @classmethod
3537
+ def DilateOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
3538
+ return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
3539
+
3540
+ # DilateOptions
3541
+ def Init(self, buf, pos):
3542
+ self._tab = flatbuffers.table.Table(buf, pos)
3543
+
3544
+ def DilateOptionsStart(builder): builder.StartObject(0)
3545
+ def Start(builder):
3546
+ return DilateOptionsStart(builder)
3547
+ def DilateOptionsEnd(builder): return builder.EndObject()
3548
+ def End(builder):
3549
+ return DilateOptionsEnd(builder)
3550
+
3551
+ class DilateOptionsT(object):
3552
+
3553
+ # DilateOptionsT
3554
+ def __init__(self):
3555
+ pass
3556
+
3557
+ @classmethod
3558
+ def InitFromBuf(cls, buf, pos):
3559
+ dilateOptions = DilateOptions()
3560
+ dilateOptions.Init(buf, pos)
3561
+ return cls.InitFromObj(dilateOptions)
3562
+
3563
+ @classmethod
3564
+ def InitFromObj(cls, dilateOptions):
3565
+ x = DilateOptionsT()
3566
+ x._UnPack(dilateOptions)
3567
+ return x
3568
+
3569
+ # DilateOptionsT
3570
+ def _UnPack(self, dilateOptions):
3571
+ if dilateOptions is None:
3572
+ return
3573
+
3574
+ # DilateOptionsT
3575
+ def Pack(self, builder):
3576
+ DilateOptionsStart(builder)
3577
+ dilateOptions = DilateOptionsEnd(builder)
3578
+ return dilateOptions
3579
+ # automatically generated by the FlatBuffers compiler, do not modify
3580
+
3581
+ # namespace: tflite
3582
+
3583
+ from flatbuffers.compat import import_numpy
3584
+ np = import_numpy()
3585
+
3518
3586
class DimensionMetadata(object):
3519
3587
__slots__ = ['_tab']
3520
3588
@@ -7775,7 +7843,7 @@ def __init__(self):
7775
7843
self.largeCustomOptionsOffset = 0 # type: int
7776
7844
self.largeCustomOptionsSize = 0 # type: int
7777
7845
self.builtinOptions2Type = 0 # type: int
7778
- self.builtinOptions2 = None # type: Union[None, StablehloConcatenateOptionsT, StablehloBroadcastInDimOptionsT, StablehloSliceOptionsT, StablehloConvolutionOptionsT, StablehloCustomCallOptionsT, StablehloReduceOptionsT, StablehloScatterOptionsT, StablehloCompareOptionsT, StablehloDynamicSliceOptionsT, StablehloPadOptionsT, StablehloIotaOptionsT, StablehloDotGeneralOptionsT, StablehloReduceWindowOptionsT, StablehloSortOptionsT, StablehloWhileOptionsT, StablehloGatherOptionsT, StablehloTransposeOptionsT]
7846
+ self.builtinOptions2 = None # type: Union[None, StablehloConcatenateOptionsT, StablehloBroadcastInDimOptionsT, StablehloSliceOptionsT, StablehloConvolutionOptionsT, StablehloCustomCallOptionsT, StablehloReduceOptionsT, StablehloScatterOptionsT, StablehloCompareOptionsT, StablehloDynamicSliceOptionsT, StablehloPadOptionsT, StablehloIotaOptionsT, StablehloDotGeneralOptionsT, StablehloReduceWindowOptionsT, StablehloSortOptionsT, StablehloWhileOptionsT, StablehloGatherOptionsT, StablehloTransposeOptionsT, DilateOptionsT ]
7779
7847
7780
7848
@classmethod
7781
7849
def InitFromBuf(cls, buf, pos):
0 commit comments