ai-edge-litert-nightly 2.2.0.dev20251231__cp39-cp39-macosx_12_0_arm64.whl

ai_edge_litert/source_context_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: google/protobuf/source_context.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'google/protobuf/source_context.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$google/protobuf/source_context.proto\x12\x0fgoogle.protobuf\"\"\n\rSourceContext\x12\x11\n\tfile_name\x18\x01 \x01(\tB\x8a\x01\n\x13\x63om.google.protobufB\x12SourceContextProtoP\x01Z6google.golang.org/protobuf/types/known/sourcecontextpb\xa2\x02\x03GPB\xaa\x02\x1eGoogle.Protobuf.WellKnownTypesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.source_context_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\023com.google.protobufB\022SourceContextProtoP\001Z6google.golang.org/protobuf/types/known/sourcecontextpb\242\002\003GPB\252\002\036Google.Protobuf.WellKnownTypes'
+  _globals['_SOURCECONTEXT']._serialized_start=57
+  _globals['_SOURCECONTEXT']._serialized_end=91
+# @@protoc_insertion_point(module_scope)

ai_edge_litert/struct_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: google/protobuf/struct.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'google/protobuf/struct.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cgoogle/protobuf/struct.proto\x12\x0fgoogle.protobuf\"\x84\x01\n\x06Struct\x12\x33\n\x06\x66ields\x18\x01 \x03(\x0b\x32#.google.protobuf.Struct.FieldsEntry\x1a\x45\n\x0b\x46ieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.Value:\x02\x38\x01\"\xea\x01\n\x05Value\x12\x30\n\nnull_value\x18\x01 \x01(\x0e\x32\x1a.google.protobuf.NullValueH\x00\x12\x16\n\x0cnumber_value\x18\x02 \x01(\x01H\x00\x12\x16\n\x0cstring_value\x18\x03 \x01(\tH\x00\x12\x14\n\nbool_value\x18\x04 \x01(\x08H\x00\x12/\n\x0cstruct_value\x18\x05 \x01(\x0b\x32\x17.google.protobuf.StructH\x00\x12\x30\n\nlist_value\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.ListValueH\x00\x42\x06\n\x04kind\"3\n\tListValue\x12&\n\x06values\x18\x01 \x03(\x0b\x32\x16.google.protobuf.Value*\x1b\n\tNullValue\x12\x0e\n\nNULL_VALUE\x10\x00\x42\x7f\n\x13\x63om.google.protobufB\x0bStructProtoP\x01Z/google.golang.org/protobuf/types/known/structpb\xf8\x01\x01\xa2\x02\x03GPB\xaa\x02\x1eGoogle.Protobuf.WellKnownTypesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.struct_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\023com.google.protobufB\013StructProtoP\001Z/google.golang.org/protobuf/types/known/structpb\370\001\001\242\002\003GPB\252\002\036Google.Protobuf.WellKnownTypes'
+  _globals['_STRUCT_FIELDSENTRY']._loaded_options = None
+  _globals['_STRUCT_FIELDSENTRY']._serialized_options = b'8\001'
+  _globals['_NULLVALUE']._serialized_start=474
+  _globals['_NULLVALUE']._serialized_end=501
+  _globals['_STRUCT']._serialized_start=50
+  _globals['_STRUCT']._serialized_end=182
+  _globals['_STRUCT_FIELDSENTRY']._serialized_start=113
+  _globals['_STRUCT_FIELDSENTRY']._serialized_end=182
+  _globals['_VALUE']._serialized_start=185
+  _globals['_VALUE']._serialized_end=419
+  _globals['_LISTVALUE']._serialized_start=421
+  _globals['_LISTVALUE']._serialized_end=472
+# @@protoc_insertion_point(module_scope)

ai_edge_litert/tensor_buffer.py

@@ -0,0 +1,167 @@
+# Copyright 2025 Google LLC.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Python wrapper for LiteRT tensor buffer."""
+
+import os
+import numpy as np
+
+# pylint: disable=g-import-not-at-top
+if not os.path.splitext(__file__)[0].endswith(
+    os.path.join("ai_edge_litert", "tensor_buffer")
+):
+  # This file is part of litert package.
+  from ai_edge_litert.litert_wrapper.tensor_buffer_wrapper import _pywrap_litert_tensor_buffer_wrapper as _tb
+else:
+  # This file is part of ai_edge_litert package.
+  from ai_edge_litert import _pywrap_litert_tensor_buffer_wrapper as _tb
+# pylint: enable=g-import-not-at-top
+
+
+class TensorBuffer:
+  """Python wrapper for LiteRtTensorBuffer.
+
+  This class provides a high-level interface to the underlying C++
+  LiteRtTensorBuffer
+  implementation, allowing for creation, reading, writing, and management of
+  tensor
+  buffers in Python.
+  """
+
+  def __init__(self, capsule):
+    """Initializes a TensorBuffer with the provided PyCapsule.
+
+    Args:
+      capsule: A PyCapsule containing a pointer to a LiteRtTensorBuffer.
+    """
+    self._capsule = capsule
+
+  @staticmethod
+  def _dtype_to_str(np_dtype):
+    """Converts a numpy dtype to a string representation.
+
+    Args:
+      np_dtype: A numpy dtype (e.g., np.float32, np.int8).
+
+    Returns:
+      String representation of the data type (e.g., "float32", "int8").
+
+    Raises:
+      ValueError: If the dtype is not supported.
+    """
+    if np_dtype == np.float32:
+      return "float32"
+    elif np_dtype == np.int32:
+      return "int32"
+    elif np_dtype == np.int8:
+      return "int8"
+    else:
+      raise ValueError(f"Unsupported dtype: {np_dtype}")
+
+  @classmethod
+  def create_from_host_memory(cls, data_array):
+    """Creates a new TensorBuffer referencing existing host memory.
+
+    Args:
+      data_array: A NumPy array (e.g., np.array([[1.0, 2.0, 3.0, 4.0]],
+        dtype=np.float32)). The dtype of the array is used.
+
+    Returns:
+      A new TensorBuffer instance.
+
+    Raises:
+      ValueError: If the input is not a NumPy array or has an unsupported dtype.
+    """
+    if not isinstance(data_array, np.ndarray):
+      raise ValueError("data_array must be a NumPy array")
+
+    dtype_str = cls._dtype_to_str(data_array.dtype)
+    num_elements = data_array.size
+
+    cap = _tb.CreateTensorBufferFromHostMemory(
+        data_array, dtype_str, num_elements
+    )
+    return cls(cap)
+
+  def write(self, data_array):
+    """Writes data to this tensor buffer.
+
+    Args:
+      data_array: A NumPy array (e.g., np.array([[1.0, 2.0, 3.0, 4.0]],
+        dtype=np.float32)). The dtype of the array is used.
+
+    Example:
+      # Using NumPy array
+      test_input = np.array([[1.0, 2.0, 3.0, 4.0]], dtype=np.float32)
+      tensor_buffer.write(test_input)
+
+    Raises:
+      ValueError: If the input is not a NumPy array or has an unsupported dtype.
+    """
+    if not isinstance(data_array, np.ndarray):
+      raise ValueError("data_array must be a NumPy array")
+
+    dtype_str = self._dtype_to_str(data_array.dtype)
+    _tb.WriteTensor(self._capsule, data_array.flatten().tolist(), dtype_str)
+
+  def read(self, num_elements: int, output_dtype):
+    """Reads data from this tensor buffer.
+
+    Args:
+      num_elements: Number of elements to read.
+      output_dtype: NumPy dtype for the output (e.g., np.float32, np.int8).
+
+    Returns:
+      A NumPy array containing the tensor data.
+
+    Example:
+      # Get output as NumPy array
+      output_array = tensor_buffer.read(4, np.float32).reshape((1, 4))
+
+    Raises:
+      ValueError: If output_dtype is not a NumPy dtype or is not supported.
+    """
+    if not isinstance(output_dtype, type) or not hasattr(
+        np, output_dtype.__name__
+    ):
+      raise ValueError(f"output_dtype must be a NumPy dtype (e.g., np.float32)")
+
+    dtype_str = self._dtype_to_str(output_dtype)
+    data_list = _tb.ReadTensor(self._capsule, num_elements, dtype_str)
+    return np.array(data_list, dtype=output_dtype)
+
+  def destroy(self):
+    """Explicitly releases resources associated with this tensor buffer.
+
+    After calling this method, the tensor buffer is no longer valid for use.
+    """
+    _tb.DestroyTensorBuffer(self._capsule)
+    self._capsule = None
+
+  @property
+  def capsule(self):
+    """Returns the underlying PyCapsule for direct C++ interoperability.
+
+    When the capsule is used with compiled_model methods, the ownership remains
+    with this TensorBuffer instance. This avoids double-free when both objects
+    try to destroy the same underlying tensor buffer.
+
+    Returns:
+      The PyCapsule containing the pointer to the LiteRtTensorBuffer, with
+      ownership remaining with this TensorBuffer instance.
+    """
+    # Once destroyed, shouldn't try to access the capsule again.
+    if self._capsule is None:
+      raise ValueError("TensorBuffer has been destroyed")
+    return self._capsule

ai_edge_litert/timestamp_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: google/protobuf/timestamp.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'google/protobuf/timestamp.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fgoogle/protobuf/timestamp.proto\x12\x0fgoogle.protobuf\"+\n\tTimestamp\x12\x0f\n\x07seconds\x18\x01 \x01(\x03\x12\r\n\x05nanos\x18\x02 \x01(\x05\x42\x85\x01\n\x13\x63om.google.protobufB\x0eTimestampProtoP\x01Z2google.golang.org/protobuf/types/known/timestamppb\xf8\x01\x01\xa2\x02\x03GPB\xaa\x02\x1eGoogle.Protobuf.WellKnownTypesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.timestamp_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\023com.google.protobufB\016TimestampProtoP\001Z2google.golang.org/protobuf/types/known/timestamppb\370\001\001\242\002\003GPB\252\002\036Google.Protobuf.WellKnownTypes'
+  _globals['_TIMESTAMP']._serialized_start=52
+  _globals['_TIMESTAMP']._serialized_end=95
+# @@protoc_insertion_point(module_scope)

ai_edge_litert/type_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: google/protobuf/type.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'google/protobuf/type.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from google.protobuf import source_context_pb2 as google_dot_protobuf_dot_source__context__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1agoogle/protobuf/type.proto\x12\x0fgoogle.protobuf\x1a\x19google/protobuf/any.proto\x1a$google/protobuf/source_context.proto\"\xe8\x01\n\x04Type\x12\x0c\n\x04name\x18\x01 \x01(\t\x12&\n\x06\x66ields\x18\x02 \x03(\x0b\x32\x16.google.protobuf.Field\x12\x0e\n\x06oneofs\x18\x03 \x03(\t\x12(\n\x07options\x18\x04 \x03(\x0b\x32\x17.google.protobuf.Option\x12\x36\n\x0esource_context\x18\x05 \x01(\x0b\x32\x1e.google.protobuf.SourceContext\x12\'\n\x06syntax\x18\x06 \x01(\x0e\x32\x17.google.protobuf.Syntax\x12\x0f\n\x07\x65\x64ition\x18\x07 \x01(\t\"\xd5\x05\n\x05\x46ield\x12)\n\x04kind\x18\x01 \x01(\x0e\x32\x1b.google.protobuf.Field.Kind\x12\x37\n\x0b\x63\x61rdinality\x18\x02 \x01(\x0e\x32\".google.protobuf.Field.Cardinality\x12\x0e\n\x06number\x18\x03 \x01(\x05\x12\x0c\n\x04name\x18\x04 \x01(\t\x12\x10\n\x08type_url\x18\x06 \x01(\t\x12\x13\n\x0boneof_index\x18\x07 \x01(\x05\x12\x0e\n\x06packed\x18\x08 \x01(\x08\x12(\n\x07options\x18\t \x03(\x0b\x32\x17.google.protobuf.Option\x12\x11\n\tjson_name\x18\n \x01(\t\x12\x15\n\rdefault_value\x18\x0b \x01(\t\"\xc8\x02\n\x04Kind\x12\x10\n\x0cTYPE_UNKNOWN\x10\x00\x12\x0f\n\x0bTYPE_DOUBLE\x10\x01\x12\x0e\n\nTYPE_FLOAT\x10\x02\x12\x0e\n\nTYPE_INT64\x10\x03\x12\x0f\n\x0bTYPE_UINT64\x10\x04\x12\x0e\n\nTYPE_INT32\x10\x05\x12\x10\n\x0cTYPE_FIXED64\x10\x06\x12\x10\n\x0cTYPE_FIXED32\x10\x07\x12\r\n\tTYPE_BOOL\x10\x08\x12\x0f\n\x0bTYPE_STRING\x10\t\x12\x0e\n\nTYPE_GROUP\x10\n\x12\x10\n\x0cTYPE_MESSAGE\x10\x0b\x12\x0e\n\nTYPE_BYTES\x10\x0c\x12\x0f\n\x0bTYPE_UINT32\x10\r\x12\r\n\tTYPE_ENUM\x10\x0e\x12\x11\n\rTYPE_SFIXED32\x10\x0f\x12\x11\n\rTYPE_SFIXED64\x10\x10\x12\x0f\n\x0bTYPE_SINT32\x10\x11\x12\x0f\n\x0bTYPE_SINT64\x10\x12\"t\n\x0b\x43\x61rdinality\x12\x17\n\x13\x43\x41RDINALITY_UNKNOWN\x10\x00\x12\x18\n\x14\x43\x41RDINALITY_OPTIONAL\x10\x01\x12\x18\n\x14\x43\x41RDINALITY_REQUIRED\x10\x02\x12\x18\n\x14\x43\x41RDINALITY_REPEATED\x10\x03\"\xdf\x01\n\x04\x45num\x12\x0c\n\x04name\x18\x01 \x01(\t\x12-\n\tenumvalue\x18\x02 \x03(\x0b\x32\x1a.google.protobuf.EnumValue\x12(\n\x07options\x18\x03 \x03(\x0b\x32\x17.google.protobuf.Option\x12\x36\n\x0esource_context\x18\x04 \x01(\x0b\x32\x1e.google.protobuf.SourceContext\x12\'\n\x06syntax\x18\x05 \x01(\x0e\x32\x17.google.protobuf.Syntax\x12\x0f\n\x07\x65\x64ition\x18\x06 \x01(\t\"S\n\tEnumValue\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06number\x18\x02 \x01(\x05\x12(\n\x07options\x18\x03 \x03(\x0b\x32\x17.google.protobuf.Option\";\n\x06Option\x12\x0c\n\x04name\x18\x01 \x01(\t\x12#\n\x05value\x18\x02 \x01(\x0b\x32\x14.google.protobuf.Any*C\n\x06Syntax\x12\x11\n\rSYNTAX_PROTO2\x10\x00\x12\x11\n\rSYNTAX_PROTO3\x10\x01\x12\x13\n\x0fSYNTAX_EDITIONS\x10\x02\x42{\n\x13\x63om.google.protobufB\tTypeProtoP\x01Z-google.golang.org/protobuf/types/known/typepb\xf8\x01\x01\xa2\x02\x03GPB\xaa\x02\x1eGoogle.Protobuf.WellKnownTypesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.type_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\023com.google.protobufB\tTypeProtoP\001Z-google.golang.org/protobuf/types/known/typepb\370\001\001\242\002\003GPB\252\002\036Google.Protobuf.WellKnownTypes'
+  _globals['_SYNTAX']._serialized_start=1447
+  _globals['_SYNTAX']._serialized_end=1514
+  _globals['_TYPE']._serialized_start=113
+  _globals['_TYPE']._serialized_end=345
+  _globals['_FIELD']._serialized_start=348
+  _globals['_FIELD']._serialized_end=1073
+  _globals['_FIELD_KIND']._serialized_start=627
+  _globals['_FIELD_KIND']._serialized_end=955
+  _globals['_FIELD_CARDINALITY']._serialized_start=957
+  _globals['_FIELD_CARDINALITY']._serialized_end=1073
+  _globals['_ENUM']._serialized_start=1076
+  _globals['_ENUM']._serialized_end=1299
+  _globals['_ENUMVALUE']._serialized_start=1301
+  _globals['_ENUMVALUE']._serialized_end=1384
+  _globals['_OPTION']._serialized_start=1386
+  _globals['_OPTION']._serialized_end=1445
+# @@protoc_insertion_point(module_scope)

ai_edge_litert/wrappers_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: google/protobuf/wrappers.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'google/protobuf/wrappers.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1egoogle/protobuf/wrappers.proto\x12\x0fgoogle.protobuf\"\x1c\n\x0b\x44oubleValue\x12\r\n\x05value\x18\x01 \x01(\x01\"\x1b\n\nFloatValue\x12\r\n\x05value\x18\x01 \x01(\x02\"\x1b\n\nInt64Value\x12\r\n\x05value\x18\x01 \x01(\x03\"\x1c\n\x0bUInt64Value\x12\r\n\x05value\x18\x01 \x01(\x04\"\x1b\n\nInt32Value\x12\r\n\x05value\x18\x01 \x01(\x05\"\x1c\n\x0bUInt32Value\x12\r\n\x05value\x18\x01 \x01(\r\"\x1a\n\tBoolValue\x12\r\n\x05value\x18\x01 \x01(\x08\"\x1c\n\x0bStringValue\x12\r\n\x05value\x18\x01 \x01(\t\"\x1b\n\nBytesValue\x12\r\n\x05value\x18\x01 \x01(\x0c\x42\x83\x01\n\x13\x63om.google.protobufB\rWrappersProtoP\x01Z1google.golang.org/protobuf/types/known/wrapperspb\xf8\x01\x01\xa2\x02\x03GPB\xaa\x02\x1eGoogle.Protobuf.WellKnownTypesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.wrappers_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\023com.google.protobufB\rWrappersProtoP\001Z1google.golang.org/protobuf/types/known/wrapperspb\370\001\001\242\002\003GPB\252\002\036Google.Protobuf.WellKnownTypes'
+  _globals['_DOUBLEVALUE']._serialized_start=51
+  _globals['_DOUBLEVALUE']._serialized_end=79
+  _globals['_FLOATVALUE']._serialized_start=81
+  _globals['_FLOATVALUE']._serialized_end=108
+  _globals['_INT64VALUE']._serialized_start=110
+  _globals['_INT64VALUE']._serialized_end=137
+  _globals['_UINT64VALUE']._serialized_start=139
+  _globals['_UINT64VALUE']._serialized_end=167
+  _globals['_INT32VALUE']._serialized_start=169
+  _globals['_INT32VALUE']._serialized_end=196
+  _globals['_UINT32VALUE']._serialized_start=198
+  _globals['_UINT32VALUE']._serialized_end=226
+  _globals['_BOOLVALUE']._serialized_start=228
+  _globals['_BOOLVALUE']._serialized_end=254
+  _globals['_STRINGVALUE']._serialized_start=256
+  _globals['_STRINGVALUE']._serialized_end=284
+  _globals['_BYTESVALUE']._serialized_start=286
+  _globals['_BYTESVALUE']._serialized_end=313
+# @@protoc_insertion_point(module_scope)

ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/METADATA

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: ai-edge-litert-nightly
+Version: 2.2.0.dev20251231
+Summary: LiteRT is for mobile and embedded devices.
+Home-page: https://www.tensorflow.org/lite/
+Author: Google AI Edge Authors
+Author-email: packages@tensorflow.org
+License: Apache 2.0
+Keywords: litert tflite tensorflow tensor machine learning
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: backports.strenum
+Requires-Dist: flatbuffers
+Requires-Dist: numpy>=1.23.2
+Requires-Dist: tqdm
+Requires-Dist: typing-extensions
+Requires-Dist: protobuf
+Provides-Extra: npu-sdk
+Requires-Dist: ai-edge-litert-sdk-qualcomm~=0.1.0; extra == "npu-sdk"
+Requires-Dist: ai-edge-litert-sdk-mediatek~=0.1.0; extra == "npu-sdk"
+Provides-Extra: model-utils
+Requires-Dist: lark; extra == "model-utils"
+Requires-Dist: ml_dtypes; extra == "model-utils"
+Requires-Dist: xdsl==0.28.0; extra == "model-utils"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: summary
+
+LiteRT is the official solution for running machine learning models on mobile
+and embedded devices. It enables on-device machine learning inference with low
+latency and a small binary size on Android, iOS, and other operating systems.

ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/RECORD

@@ -0,0 +1,38 @@
+ai_edge_litert/__init__.py,sha256=Osz-y6tS-4lmfQ2pB6_mZp6f8Ao8Rs9CRLCfr3BTjGE,34
+ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=KJWvtGideQcgmtcO5kS0CrTFd9jb5fR_3M_L86vn130,50056
+ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=AEh1dIRaoZcCXS2DfB4sOyWapI6a4aX2_hUCWPouMp4,50104
+ai_edge_litert/_pywrap_litert_interpreter_wrapper.so,sha256=iu9dJHplvfnBVPZFGESluIAZIaKUjKa9CzaQsYO5ytE,50088
+ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=baP-I7SfkfvLwGi1sHjE99_BVA8fl4QduISwimEQ4aw,50104
+ai_edge_litert/_pywrap_modify_model_interface.so,sha256=LvwwnFQn8nN3klIrM45KurxGmJCCWwsfMOdji2Zyr9w,50072
+ai_edge_litert/_pywrap_string_util.so,sha256=NvT3m46-zCTqb8bx5ROfOzSL1QxsMm8IGBllRwIk4hU,50040
+ai_edge_litert/_pywrap_tensorflow_lite_calibration_wrapper.so,sha256=eAIsFXRoCRkgHltTqdjEP9j3uaNgg-ZqhsxLgUfiGiw,50136
+ai_edge_litert/_pywrap_tensorflow_lite_metrics_wrapper.so,sha256=lIJiWOz4gulPTIMshq7Qwqf4dwWDq6LOSjwN_n8PmrQ,50104
+ai_edge_litert/any_pb2.py,sha256=W6duyvBgx7RvePFCrJSxWagU7ddj1W9l8CsjarJJPOs,1703
+ai_edge_litert/api_pb2.py,sha256=Jpn2ZFBihZFhhe47zMohsNMGIhJ9IL6VEFcLrTAlTao,2935
+ai_edge_litert/compiled_model.py,sha256=xjoYYk5vHU6jP1QqjiLG5y7I5JpnBNOkl9khe6IT-Pw,7933
+ai_edge_litert/descriptor_pb2.py,sha256=H0UB_-Xu4emXuqjimUeo6BaVsI0bkRex0ZWdEteBBRA,355639
+ai_edge_litert/duration_pb2.py,sha256=ot6pGJwRMjEoJE0u-uoXmzqIjCLVVNb3vLSKhsf1w2c,1783
+ai_edge_litert/empty_pb2.py,sha256=RtYfjWZz9WAe3fY5vDhjSo3i0M6kwZ8-kX9R_9vmyGc,1669
+ai_edge_litert/field_mask_pb2.py,sha256=YVi1IasKjEQAlcIwM6HMgmW0wT6TMAXjiaiysrEY7Gc,1756
+ai_edge_litert/format_converter_wrapper_pybind11.so,sha256=ejkl6_Invdc8WZn6Rz5jsBHPCkWl6roa8w4DN7Hf1Bg,50088
+ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
+ai_edge_litert/interpreter.py,sha256=u3QCfoVGbMVhVRu9w6FtWhyHEsyMRdBAvUK4u0ts2L4,40699
+ai_edge_litert/libLiteRt.so,sha256=JTqrnboffT5OUa0JhfvfuUlFTbS4mkAuWVO4YBUbduw,5561008
+ai_edge_litert/libpywrap_litert_common.dylib,sha256=8i_jumlZU64XbUeuqLqolcDbPxxG4hIPDArIhdzt8oE,11352640
+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=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=LdTM1ts_w43vyHPg4zY8xbZ61f-AtjYJC5yJB2R_85I,50024
+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.2.0.dev20251231.dist-info/METADATA,sha256=_vyjyR1GWfr_8CLBAWjfdDoHhvgUpucemAAXsKIF0gI,2108
+ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/WHEEL,sha256=aucntbTc6bEPmSzCHROrNYAuYGC7v9_aTX954jHchmw,107
+ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/RECORD,,

ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (78.1.1)
+Root-Is-Purelib: false
+Tag: cp39-cp39-macosx_12_0_arm64
+

ai_edge_litert_nightly-2.2.0.dev20251231.dist-info/top_level.txt

@@ -0,0 +1 @@
+ai_edge_litert