ai-edge-litert-nightly 2.0.4.dev20251013__cp311-cp311-macosx_12_0_arm64.whl → 2.0.4.dev20251107__cp311-cp311-macosx_12_0_arm64.whl

ai_edge_litert/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.4.dev20251013"
+__version__ = "2.0.4.dev20251107"

ai_edge_litert/model_runtime_info_pb2.py

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 from ai_edge_litert import profiling_info_pb2 as tflite_dot_profiling_dot_proto_dot_profiling__info__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/tflite/profiling/proto/model_runtime_info.proto\x12\x10tflite.profiling\x1a+tflite/profiling/proto/profiling_info.proto\"_\n\x13ModelRuntimeDetails\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12\x34\n\tsubgraphs\x18\x02 \x03(\x0b\x32!.tflite.profiling.RuntimeSubgraph\"\xb7\x02\n\x0fRuntimeSubgraph\x12\x13\n\x0bsubgraph_id\x18\x01 \x01(\x05\x12%\n\x05\x65\x64ges\x18\x02 \x03(\x0b\x32\x16.tflite.profiling.Edge\x12%\n\x05nodes\x18\x03 \x03(\x0b\x32\x16.tflite.profiling.Node\x12\x1a\n\x0e\x65xecution_plan\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x45\n\rsubgraph_type\x18\x05 \x01(\x0e\x32..tflite.profiling.RuntimeSubgraph.SubgraphType\x12\x0c\n\x04name\x18\x06 \x01(\t\"P\n\x0cSubgraphType\x12\x14\n\x10UNKNOWN_SUBGRAPH\x10\x00\x12\x13\n\x0fTFLITE_SUBGRAPH\x10\x01\x12\x15\n\x11\x44\x45LEGATE_SUBGRAPH\x10\x02\"\xba\x02\n\x04Node\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04type\x18\x03 \x01(\t\x12\x12\n\x06inputs\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x13\n\x07outputs\x18\x05 \x03(\x05\x42\x02\x10\x01\x12\x19\n\rintermediates\x18\x06 \x03(\x05\x42\x02\x10\x01\x12\x17\n\x0btemporaries\x18\x07 \x03(\x05\x42\x02\x10\x01\x12\x38\n\x0fop_profile_data\x18\n \x01(\x0b\x32\x1f.tflite.profiling.OpProfileData\x12\x46\n\x15\x64\x65legate_node_details\x18\x08 \x01(\x0b\x32%.tflite.profiling.DelegateNodeDetailsH\x00\x12\x1e\n\x14\x64\x65legated_to_node_id\x18\t \x01(\x05H\x00\x42\x0b\n\tnode_info\"R\n\x13\x44\x65legateNodeDetails\x12\x15\n\rdelegate_name\x18\x01 \x01(\t\x12$\n\x18tflite_node_ids_replaced\x18\x02 \x03(\x05\x42\x02\x10\x01\"\x81\x05\n\x04\x45\x64ge\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x32\n\tdata_type\x18\x03 \x01(\x0e\x32\x1f.tflite.profiling.Edge.DataType\x12\x11\n\x05shape\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x17\n\x0f\x61llocation_type\x18\x05 \x01(\t\x12\x36\n\x0blayout_type\x18\x06 \x01(\x0e\x32!.tflite.profiling.Edge.LayoutType\x12\x0c\n\x04size\x18\x07 \x01(\x05\"\x85\x02\n\x08\x44\x61taType\x12\x10\n\x0cUNKNOWN_TYPE\x10\x00\x12\x0b\n\x07\x46LOAT32\x10\x01\x12\t\n\x05INT32\x10\x02\x12\t\n\x05UINT8\x10\x03\x12\t\n\x05INT64\x10\x04\x12\n\n\x06STRING\x10\x05\x12\x08\n\x04\x42OOL\x10\x06\x12\t\n\x05INT16\x10\x07\x12\r\n\tCOMPLEX64\x10\x08\x12\x08\n\x04INT8\x10\t\x12\x0b\n\x07\x46LOAT16\x10\n\x12\x0b\n\x07\x46LOAT64\x10\x0b\x12\x0e\n\nCOMPLEX128\x10\x0c\x12\n\n\x06UINT64\x10\r\x12\x0c\n\x08RESOURCE\x10\x0e\x12\x0b\n\x07VARIANT\x10\x0f\x12\n\n\x06UINT32\x10\x10\x12\n\n\x06UINT16\x10\x11\x12\x08\n\x04INT4\x10\x12\x12\x0c\n\x08\x42\x46LOAT16\x10\x13\"\xb0\x01\n\nLayoutType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\n\n\x06SCALAR\x10\x01\x12\n\n\x06LINEAR\x10\x02\x12\x06\n\x02HW\x10\x03\x12\x07\n\x03\x43HW\x10\x04\x12\x07\n\x03HWC\x10\x05\x12\x08\n\x04OIHW\x10\x06\x12\x08\n\x04OHWI\x10\x07\x12\x08\n\x04IHWO\x10\x08\x12\x08\n\x04IOHW\x10\t\x12\x08\n\x04\x42HWC\x10\n\x12\x08\n\x04HWDC\x10\x0b\x12\t\n\x05\x42HWDC\x10\x0c\x12\x07\n\x03HWD\x10\r\x12\t\n\x05OHWDI\x10\x0e\x12\x08\n\x04HWIO\x10\x0f\x42\x14\n\x10tflite.profilingP\x01')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/tflite/profiling/proto/model_runtime_info.proto\x12\x10tflite.profiling\x1a+tflite/profiling/proto/profiling_info.proto\"_\n\x13ModelRuntimeDetails\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12\x34\n\tsubgraphs\x18\x02 \x03(\x0b\x32!.tflite.profiling.RuntimeSubgraph\"\xb7\x02\n\x0fRuntimeSubgraph\x12\x13\n\x0bsubgraph_id\x18\x01 \x01(\x05\x12%\n\x05\x65\x64ges\x18\x02 \x03(\x0b\x32\x16.tflite.profiling.Edge\x12%\n\x05nodes\x18\x03 \x03(\x0b\x32\x16.tflite.profiling.Node\x12\x1a\n\x0e\x65xecution_plan\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x45\n\rsubgraph_type\x18\x05 \x01(\x0e\x32..tflite.profiling.RuntimeSubgraph.SubgraphType\x12\x0c\n\x04name\x18\x06 \x01(\t\"P\n\x0cSubgraphType\x12\x14\n\x10UNKNOWN_SUBGRAPH\x10\x00\x12\x13\n\x0fTFLITE_SUBGRAPH\x10\x01\x12\x15\n\x11\x44\x45LEGATE_SUBGRAPH\x10\x02\"\xba\x02\n\x04Node\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04type\x18\x03 \x01(\t\x12\x12\n\x06inputs\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x13\n\x07outputs\x18\x05 \x03(\x05\x42\x02\x10\x01\x12\x19\n\rintermediates\x18\x06 \x03(\x05\x42\x02\x10\x01\x12\x17\n\x0btemporaries\x18\x07 \x03(\x05\x42\x02\x10\x01\x12\x38\n\x0fop_profile_data\x18\n \x01(\x0b\x32\x1f.tflite.profiling.OpProfileData\x12\x46\n\x15\x64\x65legate_node_details\x18\x08 \x01(\x0b\x32%.tflite.profiling.DelegateNodeDetailsH\x00\x12\x1e\n\x14\x64\x65legated_to_node_id\x18\t \x01(\x05H\x00\x42\x0b\n\tnode_info\"R\n\x13\x44\x65legateNodeDetails\x12\x15\n\rdelegate_name\x18\x01 \x01(\t\x12$\n\x18tflite_node_ids_replaced\x18\x02 \x03(\x05\x42\x02\x10\x01\"\x8b\x05\n\x04\x45\x64ge\x12\n\n\x02id\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x32\n\tdata_type\x18\x03 \x01(\x0e\x32\x1f.tflite.profiling.Edge.DataType\x12\x11\n\x05shape\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x17\n\x0f\x61llocation_type\x18\x05 \x01(\t\x12\x36\n\x0blayout_type\x18\x06 \x01(\x0e\x32!.tflite.profiling.Edge.LayoutType\x12\x0c\n\x04size\x18\x07 \x01(\x05\"\x8f\x02\n\x08\x44\x61taType\x12\x10\n\x0cUNKNOWN_TYPE\x10\x00\x12\x0b\n\x07\x46LOAT32\x10\x01\x12\t\n\x05INT32\x10\x02\x12\t\n\x05UINT8\x10\x03\x12\t\n\x05INT64\x10\x04\x12\n\n\x06STRING\x10\x05\x12\x08\n\x04\x42OOL\x10\x06\x12\t\n\x05INT16\x10\x07\x12\r\n\tCOMPLEX64\x10\x08\x12\x08\n\x04INT8\x10\t\x12\x0b\n\x07\x46LOAT16\x10\n\x12\x0b\n\x07\x46LOAT64\x10\x0b\x12\x0e\n\nCOMPLEX128\x10\x0c\x12\n\n\x06UINT64\x10\r\x12\x0c\n\x08RESOURCE\x10\x0e\x12\x0b\n\x07VARIANT\x10\x0f\x12\n\n\x06UINT32\x10\x10\x12\n\n\x06UINT16\x10\x11\x12\x08\n\x04INT4\x10\x12\x12\x0c\n\x08\x42\x46LOAT16\x10\x13\x12\x08\n\x04INT2\x10\x14\"\xb0\x01\n\nLayoutType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\n\n\x06SCALAR\x10\x01\x12\n\n\x06LINEAR\x10\x02\x12\x06\n\x02HW\x10\x03\x12\x07\n\x03\x43HW\x10\x04\x12\x07\n\x03HWC\x10\x05\x12\x08\n\x04OIHW\x10\x06\x12\x08\n\x04OHWI\x10\x07\x12\x08\n\x04IHWO\x10\x08\x12\x08\n\x04IOHW\x10\t\x12\x08\n\x04\x42HWC\x10\n\x12\x08\n\x04HWDC\x10\x0b\x12\t\n\x05\x42HWDC\x10\x0c\x12\x07\n\x03HWD\x10\r\x12\t\n\x05OHWDI\x10\x0e\x12\x08\n\x04HWIO\x10\x0f\x42\x14\n\x10tflite.profilingP\x01')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -58,9 +58,9 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_DELEGATENODEDETAILS']._serialized_start=842
   _globals['_DELEGATENODEDETAILS']._serialized_end=924
   _globals['_EDGE']._serialized_start=927
-  _globals['_EDGE']._serialized_end=1568
+  _globals['_EDGE']._serialized_end=1578
   _globals['_EDGE_DATATYPE']._serialized_start=1128
-  _globals['_EDGE_DATATYPE']._serialized_end=1389
-  _globals['_EDGE_LAYOUTTYPE']._serialized_start=1392
-  _globals['_EDGE_LAYOUTTYPE']._serialized_end=1568
+  _globals['_EDGE_DATATYPE']._serialized_end=1399
+  _globals['_EDGE_LAYOUTTYPE']._serialized_start=1402
+  _globals['_EDGE_LAYOUTTYPE']._serialized_end=1578
 # @@protoc_insertion_point(module_scope)

ai_edge_litert/schema_py_generated.py

@@ -27,6 +27,7 @@ class TensorType(object):
     UINT16 = 16
     INT4 = 17
     BFLOAT16 = 18
+    INT2 = 19
 
 
 class QuantizationDetails(object):
@@ -2450,8 +2451,15 @@ class Tensor(object):
         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
         return o == 0
 
+    # Tensor
+    def ExternalBuffer(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+        return 0
+
 def TensorStart(builder):
-    builder.StartObject(10)
+    builder.StartObject(11)
 
 def TensorAddShape(builder, shape):
     builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0)
@@ -2492,6 +2500,9 @@ def TensorAddVariantTensors(builder, variantTensors):
 def TensorStartVariantTensorsVector(builder, numElems):
     return builder.StartVector(4, numElems, 4)
 
+def TensorAddExternalBuffer(builder, externalBuffer):
+    builder.PrependUint32Slot(10, externalBuffer, 0)
+
 def TensorEnd(builder):
     return builder.EndObject()
 
@@ -2516,6 +2527,7 @@ class TensorT(object):
         shapeSignature = None,
         hasRank = False,
         variantTensors = None,
+        externalBuffer = 0,
     ):
         self.shape = shape  # type: Optional[List[int]]
         self.type = type  # type: int
@@ -2527,6 +2539,7 @@ class TensorT(object):
         self.shapeSignature = shapeSignature  # type: Optional[List[int]]
         self.hasRank = hasRank  # type: bool
         self.variantTensors = variantTensors  # type: Optional[List[VariantSubTypeT]]
+        self.externalBuffer = externalBuffer  # type: int
 
     @classmethod
     def InitFromBuf(cls, buf, pos):
@@ -2580,6 +2593,7 @@ class TensorT(object):
                 else:
                     variantSubType_ = VariantSubTypeT.InitFromObj(tensor.VariantTensors(i))
                     self.variantTensors.append(variantSubType_)
+        self.externalBuffer = tensor.ExternalBuffer()
 
     # TensorT
     def Pack(self, builder):
@@ -2630,6 +2644,7 @@ class TensorT(object):
         TensorAddHasRank(builder, self.hasRank)
         if self.variantTensors is not None:
             TensorAddVariantTensors(builder, variantTensors)
+        TensorAddExternalBuffer(builder, self.externalBuffer)
         tensor = TensorEnd(builder)
         return tensor
 
@@ -18503,6 +18518,228 @@ class BufferT(object):
         return buffer
 
 
+class ExternalBufferGroup(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = ExternalBufferGroup()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsExternalBufferGroup(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def ExternalBufferGroupBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
+    # ExternalBufferGroup
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # ExternalBufferGroup
+    def Name(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
+def ExternalBufferGroupStart(builder):
+    builder.StartObject(1)
+
+def ExternalBufferGroupAddName(builder, name):
+    builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
+
+def ExternalBufferGroupEnd(builder):
+    return builder.EndObject()
+
+
+
+class ExternalBufferGroupT(object):
+
+    # ExternalBufferGroupT
+    def __init__(
+        self,
+        name = None,
+    ):
+        self.name = name  # type: Optional[str]
+
+    @classmethod
+    def InitFromBuf(cls, buf, pos):
+        externalBufferGroup = ExternalBufferGroup()
+        externalBufferGroup.Init(buf, pos)
+        return cls.InitFromObj(externalBufferGroup)
+
+    @classmethod
+    def InitFromPackedBuf(cls, buf, pos=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
+        return cls.InitFromBuf(buf, pos+n)
+
+    @classmethod
+    def InitFromObj(cls, externalBufferGroup):
+        x = ExternalBufferGroupT()
+        x._UnPack(externalBufferGroup)
+        return x
+
+    # ExternalBufferGroupT
+    def _UnPack(self, externalBufferGroup):
+        if externalBufferGroup is None:
+            return
+        self.name = externalBufferGroup.Name()
+
+    # ExternalBufferGroupT
+    def Pack(self, builder):
+        if self.name is not None:
+            name = builder.CreateString(self.name)
+        ExternalBufferGroupStart(builder)
+        if self.name is not None:
+            ExternalBufferGroupAddName(builder, name)
+        externalBufferGroup = ExternalBufferGroupEnd(builder)
+        return externalBufferGroup
+
+
+class ExternalBuffer(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = ExternalBuffer()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsExternalBuffer(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def ExternalBufferBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
+    # ExternalBuffer
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # ExternalBuffer
+    def Id(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+        return 0
+
+    # ExternalBuffer
+    def Group(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Uint32Flags, o + self._tab.Pos)
+        return 0
+
+    # ExternalBuffer
+    def Offset(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
+        return 0
+
+    # ExternalBuffer
+    def Length(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Uint64Flags, o + self._tab.Pos)
+        return 0
+
+    # ExternalBuffer
+    def Packing(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
+def ExternalBufferStart(builder):
+    builder.StartObject(5)
+
+def ExternalBufferAddId(builder, id):
+    builder.PrependUint32Slot(0, id, 0)
+
+def ExternalBufferAddGroup(builder, group):
+    builder.PrependUint32Slot(1, group, 0)
+
+def ExternalBufferAddOffset(builder, offset):
+    builder.PrependUint64Slot(2, offset, 0)
+
+def ExternalBufferAddLength(builder, length):
+    builder.PrependUint64Slot(3, length, 0)
+
+def ExternalBufferAddPacking(builder, packing):
+    builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(packing), 0)
+
+def ExternalBufferEnd(builder):
+    return builder.EndObject()
+
+
+
+class ExternalBufferT(object):
+
+    # ExternalBufferT
+    def __init__(
+        self,
+        id = 0,
+        group = 0,
+        offset = 0,
+        length = 0,
+        packing = None,
+    ):
+        self.id = id  # type: int
+        self.group = group  # type: int
+        self.offset = offset  # type: int
+        self.length = length  # type: int
+        self.packing = packing  # type: Optional[str]
+
+    @classmethod
+    def InitFromBuf(cls, buf, pos):
+        externalBuffer = ExternalBuffer()
+        externalBuffer.Init(buf, pos)
+        return cls.InitFromObj(externalBuffer)
+
+    @classmethod
+    def InitFromPackedBuf(cls, buf, pos=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
+        return cls.InitFromBuf(buf, pos+n)
+
+    @classmethod
+    def InitFromObj(cls, externalBuffer):
+        x = ExternalBufferT()
+        x._UnPack(externalBuffer)
+        return x
+
+    # ExternalBufferT
+    def _UnPack(self, externalBuffer):
+        if externalBuffer is None:
+            return
+        self.id = externalBuffer.Id()
+        self.group = externalBuffer.Group()
+        self.offset = externalBuffer.Offset()
+        self.length = externalBuffer.Length()
+        self.packing = externalBuffer.Packing()
+
+    # ExternalBufferT
+    def Pack(self, builder):
+        if self.packing is not None:
+            packing = builder.CreateString(self.packing)
+        ExternalBufferStart(builder)
+        ExternalBufferAddId(builder, self.id)
+        ExternalBufferAddGroup(builder, self.group)
+        ExternalBufferAddOffset(builder, self.offset)
+        ExternalBufferAddLength(builder, self.length)
+        if self.packing is not None:
+            ExternalBufferAddPacking(builder, packing)
+        externalBuffer = ExternalBufferEnd(builder)
+        return externalBuffer
+
+
 class Metadata(object):
     __slots__ = ['_tab']
 
@@ -19081,8 +19318,56 @@ class Model(object):
         o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
         return o == 0
 
+    # Model
+    def ExternalBufferGroups(self, j):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
+        if o != 0:
+            x = self._tab.Vector(o)
+            x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
+            x = self._tab.Indirect(x)
+            obj = ExternalBufferGroup()
+            obj.Init(self._tab.Bytes, x)
+            return obj
+        return None
+
+    # Model
+    def ExternalBufferGroupsLength(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
+        if o != 0:
+            return self._tab.VectorLen(o)
+        return 0
+
+    # Model
+    def ExternalBufferGroupsIsNone(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
+        return o == 0
+
+    # Model
+    def ExternalBuffers(self, j):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
+        if o != 0:
+            x = self._tab.Vector(o)
+            x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
+            x = self._tab.Indirect(x)
+            obj = ExternalBuffer()
+            obj.Init(self._tab.Bytes, x)
+            return obj
+        return None
+
+    # Model
+    def ExternalBuffersLength(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
+        if o != 0:
+            return self._tab.VectorLen(o)
+        return 0
+
+    # Model
+    def ExternalBuffersIsNone(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
+        return o == 0
+
 def ModelStart(builder):
-    builder.StartObject(8)
+    builder.StartObject(10)
 
 def ModelAddVersion(builder, version):
     builder.PrependUint32Slot(0, version, 0)
@@ -19126,6 +19411,18 @@ def ModelAddSignatureDefs(builder, signatureDefs):
 def ModelStartSignatureDefsVector(builder, numElems):
     return builder.StartVector(4, numElems, 4)
 
+def ModelAddExternalBufferGroups(builder, externalBufferGroups):
+    builder.PrependUOffsetTRelativeSlot(8, flatbuffers.number_types.UOffsetTFlags.py_type(externalBufferGroups), 0)
+
+def ModelStartExternalBufferGroupsVector(builder, numElems):
+    return builder.StartVector(4, numElems, 4)
+
+def ModelAddExternalBuffers(builder, externalBuffers):
+    builder.PrependUOffsetTRelativeSlot(9, flatbuffers.number_types.UOffsetTFlags.py_type(externalBuffers), 0)
+
+def ModelStartExternalBuffersVector(builder, numElems):
+    return builder.StartVector(4, numElems, 4)
+
 def ModelEnd(builder):
     return builder.EndObject()
 
@@ -19148,6 +19445,8 @@ class ModelT(object):
         metadataBuffer = None,
         metadata = None,
         signatureDefs = None,
+        externalBufferGroups = None,
+        externalBuffers = None,
     ):
         self.version = version  # type: int
         self.operatorCodes = operatorCodes  # type: Optional[List[OperatorCodeT]]
@@ -19157,6 +19456,8 @@ class ModelT(object):
         self.metadataBuffer = metadataBuffer  # type: Optional[List[int]]
         self.metadata = metadata  # type: Optional[List[MetadataT]]
         self.signatureDefs = signatureDefs  # type: Optional[List[SignatureDefT]]
+        self.externalBufferGroups = externalBufferGroups  # type: Optional[List[ExternalBufferGroupT]]
+        self.externalBuffers = externalBuffers  # type: Optional[List[ExternalBufferT]]
 
     @classmethod
     def InitFromBuf(cls, buf, pos):
@@ -19228,6 +19529,22 @@ class ModelT(object):
                 else:
                     signatureDef_ = SignatureDefT.InitFromObj(model.SignatureDefs(i))
                     self.signatureDefs.append(signatureDef_)
+        if not model.ExternalBufferGroupsIsNone():
+            self.externalBufferGroups = []
+            for i in range(model.ExternalBufferGroupsLength()):
+                if model.ExternalBufferGroups(i) is None:
+                    self.externalBufferGroups.append(None)
+                else:
+                    externalBufferGroup_ = ExternalBufferGroupT.InitFromObj(model.ExternalBufferGroups(i))
+                    self.externalBufferGroups.append(externalBufferGroup_)
+        if not model.ExternalBuffersIsNone():
+            self.externalBuffers = []
+            for i in range(model.ExternalBuffersLength()):
+                if model.ExternalBuffers(i) is None:
+                    self.externalBuffers.append(None)
+                else:
+                    externalBuffer_ = ExternalBufferT.InitFromObj(model.ExternalBuffers(i))
+                    self.externalBuffers.append(externalBuffer_)
 
     # ModelT
     def Pack(self, builder):
@@ -19281,6 +19598,22 @@ class ModelT(object):
             for i in reversed(range(len(self.signatureDefs))):
                 builder.PrependUOffsetTRelative(signatureDefslist[i])
             signatureDefs = builder.EndVector()
+        if self.externalBufferGroups is not None:
+            externalBufferGroupslist = []
+            for i in range(len(self.externalBufferGroups)):
+                externalBufferGroupslist.append(self.externalBufferGroups[i].Pack(builder))
+            ModelStartExternalBufferGroupsVector(builder, len(self.externalBufferGroups))
+            for i in reversed(range(len(self.externalBufferGroups))):
+                builder.PrependUOffsetTRelative(externalBufferGroupslist[i])
+            externalBufferGroups = builder.EndVector()
+        if self.externalBuffers is not None:
+            externalBufferslist = []
+            for i in range(len(self.externalBuffers)):
+                externalBufferslist.append(self.externalBuffers[i].Pack(builder))
+            ModelStartExternalBuffersVector(builder, len(self.externalBuffers))
+            for i in reversed(range(len(self.externalBuffers))):
+                builder.PrependUOffsetTRelative(externalBufferslist[i])
+            externalBuffers = builder.EndVector()
         ModelStart(builder)
         ModelAddVersion(builder, self.version)
         if self.operatorCodes is not None:
@@ -19297,6 +19630,10 @@ class ModelT(object):
             ModelAddMetadata(builder, metadata)
         if self.signatureDefs is not None:
             ModelAddSignatureDefs(builder, signatureDefs)
+        if self.externalBufferGroups is not None:
+            ModelAddExternalBufferGroups(builder, externalBufferGroups)
+        if self.externalBuffers is not None:
+            ModelAddExternalBuffers(builder, externalBuffers)
         model = ModelEnd(builder)
         return model
 

{ai_edge_litert_nightly-2.0.4.dev20251013.dist-info → ai_edge_litert_nightly-2.0.4.dev20251107.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-edge-litert-nightly
-Version: 2.0.4.dev20251013
+Version: 2.0.4.dev20251107
 Summary: LiteRT is for mobile and embedded devices.
 Home-page: https://www.tensorflow.org/lite/
 Author: Google AI Edge Authors

{ai_edge_litert_nightly-2.0.4.dev20251013.dist-info → ai_edge_litert_nightly-2.0.4.dev20251107.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ai_edge_litert/__init__.py,sha256=3-rWJ0iGMyrscAzXfzR_Q8V6bpRYRMk6fSvCiV65imo,34
+ai_edge_litert/__init__.py,sha256=ZipBknKez_ouaH-nbD5QZhDt_NyJ2yk2ap6h-CBBK50,34
 ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=b208vF3tQRJUayfGJCiNN-h0JFAQHCkWx3mnQKGKSNY,50056
 ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=R48bfhjg5-oSz8DUqDfvpztzk071ZF-snPbKHzEZMow,50104
 ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=EZqmZDelToDS8vRG86Hekg2hG1pIrSMdcPtgiL6VIBg,50104
@@ -19,18 +19,18 @@ ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5
 ai_edge_litert/interpreter.py,sha256=VU2Hzn60P3vqxdYAZONLoxLEIZKbHhIF8FhUshm3uKU,40707
 ai_edge_litert/metrics_interface.py,sha256=dVu6SmbnQUntPgE5o6BxHVMyemwli-7F6tDfVMGrlYI,1542
 ai_edge_litert/metrics_portable.py,sha256=KKvR9ZOe8j2ZeBtDo_6gWJ8kENKoOawPK3LPkevnZa8,2039
-ai_edge_litert/model_runtime_info_pb2.py,sha256=E93kYJtWnsChrdegZJbKzeFpplssBGEURTdOOfjtOxg,6370
+ai_edge_litert/model_runtime_info_pb2.py,sha256=iNaM8G9qxWm3LpFTacl0KP19CJxA3_N8oeDg4d-BPx8,6396
 ai_edge_litert/plugin_pb2.py,sha256=dDsvFbuWV2yq4ghU6XnHMW6ZrbxYG2l8DAArIoqL8PY,3514
 ai_edge_litert/profiling_info_pb2.py,sha256=RxZo_P9siirO7ktyZImOxVKMZ7g-ienzuP7rSsF61YM,3579
 ai_edge_litert/pywrap_genai_ops.so,sha256=jmpe56cqSgt1rFR-4mVysQmCvSfRGzg3o1Ml6_Wt-os,50024
-ai_edge_litert/schema_py_generated.py,sha256=Sbu3LV4WhuFezr-SVLfqkh_FBsclQnOXu4ZIIVL9BF4,664149
+ai_edge_litert/schema_py_generated.py,sha256=9j4-NNxuJxeohqhYAVMgKKD9mMJsKaki1xNeguM9I6Y,676250
 ai_edge_litert/source_context_pb2.py,sha256=m_GMQpv9LUK-7oKg1Vv64fTL6Bd3ZThH-fVVLtPxdqI,1775
 ai_edge_litert/struct_pb2.py,sha256=DJ6P0fZe53upcah9J1jDo6BSClobouW9dHdCTItZO1A,2955
 ai_edge_litert/tensor_buffer.py,sha256=3Xy7kd_aR499QrfO0YITvr_z-aU1eONga41KiDBUeew,5356
 ai_edge_litert/timestamp_pb2.py,sha256=K7Gs_qOn2XAMZTNUUrJ1XcGf4OmYhdt858I_zDtZYzg,1793
 ai_edge_litert/type_pb2.py,sha256=x8rTulmlk9FxtRJal7BnUsaPbn2TcLlu8D7AVoVPjMA,5065
 ai_edge_litert/wrappers_pb2.py,sha256=4hQAZRGeaE5DyY6YQ7VfrqozPa_d_LCOBEYjxfQNlKs,2955
-ai_edge_litert_nightly-2.0.4.dev20251013.dist-info/METADATA,sha256=r_aiyTfhStnykUb0EqJrwpsjo9OMjyOVjkWjZQkV54U,1935
-ai_edge_litert_nightly-2.0.4.dev20251013.dist-info/WHEEL,sha256=GDRGfrEjbzqcqlw5Z6R4erzAGrMyWjJmj9fqgVLp_Kc,109
-ai_edge_litert_nightly-2.0.4.dev20251013.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
-ai_edge_litert_nightly-2.0.4.dev20251013.dist-info/RECORD,,
+ai_edge_litert_nightly-2.0.4.dev20251107.dist-info/METADATA,sha256=tdx1omYKNAN5UFLO-GnQ70m3JqNDL8JOCd_M6oB4q3E,1935
+ai_edge_litert_nightly-2.0.4.dev20251107.dist-info/WHEEL,sha256=GDRGfrEjbzqcqlw5Z6R4erzAGrMyWjJmj9fqgVLp_Kc,109
+ai_edge_litert_nightly-2.0.4.dev20251107.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-2.0.4.dev20251107.dist-info/RECORD,,