ai-edge-litert-nightly 2.0.4.dev20251102__cp313-cp313-macosx_12_0_arm64.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.

Files changed (36) hide show
  1. ai_edge_litert/__init__.py +1 -0
  2. ai_edge_litert/_pywrap_analyzer_wrapper.so +0 -0
  3. ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so +0 -0
  4. ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so +0 -0
  5. ai_edge_litert/_pywrap_modify_model_interface.so +0 -0
  6. ai_edge_litert/_pywrap_string_util.so +0 -0
  7. ai_edge_litert/_pywrap_tensorflow_interpreter_wrapper.so +0 -0
  8. ai_edge_litert/_pywrap_tensorflow_lite_calibration_wrapper.so +0 -0
  9. ai_edge_litert/_pywrap_tensorflow_lite_metrics_wrapper.so +0 -0
  10. ai_edge_litert/any_pb2.py +37 -0
  11. ai_edge_litert/api_pb2.py +43 -0
  12. ai_edge_litert/compiled_model.py +250 -0
  13. ai_edge_litert/descriptor_pb2.py +3361 -0
  14. ai_edge_litert/duration_pb2.py +37 -0
  15. ai_edge_litert/empty_pb2.py +37 -0
  16. ai_edge_litert/field_mask_pb2.py +37 -0
  17. ai_edge_litert/format_converter_wrapper_pybind11.so +0 -0
  18. ai_edge_litert/hardware_accelerator.py +22 -0
  19. ai_edge_litert/interpreter.py +1039 -0
  20. ai_edge_litert/metrics_interface.py +48 -0
  21. ai_edge_litert/metrics_portable.py +70 -0
  22. ai_edge_litert/model_runtime_info_pb2.py +66 -0
  23. ai_edge_litert/plugin_pb2.py +46 -0
  24. ai_edge_litert/profiling_info_pb2.py +47 -0
  25. ai_edge_litert/pywrap_genai_ops.so +0 -0
  26. ai_edge_litert/schema_py_generated.py +19640 -0
  27. ai_edge_litert/source_context_pb2.py +37 -0
  28. ai_edge_litert/struct_pb2.py +47 -0
  29. ai_edge_litert/tensor_buffer.py +167 -0
  30. ai_edge_litert/timestamp_pb2.py +37 -0
  31. ai_edge_litert/type_pb2.py +53 -0
  32. ai_edge_litert/wrappers_pb2.py +53 -0
  33. ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/METADATA +48 -0
  34. ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/RECORD +36 -0
  35. ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/WHEEL +5 -0
  36. ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/top_level.txt +1 -0
@@ -0,0 +1,37 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: google/protobuf/source_context.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 31,
16
+ 1,
17
+ '',
18
+ 'google/protobuf/source_context.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ 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')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.source_context_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _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'
35
+ _globals['_SOURCECONTEXT']._serialized_start=57
36
+ _globals['_SOURCECONTEXT']._serialized_end=91
37
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,47 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: google/protobuf/struct.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 31,
16
+ 1,
17
+ '',
18
+ 'google/protobuf/struct.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ 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')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.struct_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _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'
35
+ _globals['_STRUCT_FIELDSENTRY']._loaded_options = None
36
+ _globals['_STRUCT_FIELDSENTRY']._serialized_options = b'8\001'
37
+ _globals['_NULLVALUE']._serialized_start=474
38
+ _globals['_NULLVALUE']._serialized_end=501
39
+ _globals['_STRUCT']._serialized_start=50
40
+ _globals['_STRUCT']._serialized_end=182
41
+ _globals['_STRUCT_FIELDSENTRY']._serialized_start=113
42
+ _globals['_STRUCT_FIELDSENTRY']._serialized_end=182
43
+ _globals['_VALUE']._serialized_start=185
44
+ _globals['_VALUE']._serialized_end=419
45
+ _globals['_LISTVALUE']._serialized_start=421
46
+ _globals['_LISTVALUE']._serialized_end=472
47
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,167 @@
1
+ # Copyright 2025 Google LLC.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ """Python wrapper for LiteRT tensor buffer."""
16
+
17
+ import os
18
+ import numpy as np
19
+
20
+ # pylint: disable=g-import-not-at-top
21
+ if not os.path.splitext(__file__)[0].endswith(
22
+ os.path.join("ai_edge_litert", "tensor_buffer")
23
+ ):
24
+ # This file is part of litert package.
25
+ from ai_edge_litert.litert_wrapper.tensor_buffer_wrapper import _pywrap_litert_tensor_buffer_wrapper as _tb
26
+ else:
27
+ # This file is part of ai_edge_litert package.
28
+ from ai_edge_litert import _pywrap_litert_tensor_buffer_wrapper as _tb
29
+ # pylint: enable=g-import-not-at-top
30
+
31
+
32
+ class TensorBuffer:
33
+ """Python wrapper for LiteRtTensorBuffer.
34
+
35
+ This class provides a high-level interface to the underlying C++
36
+ LiteRtTensorBuffer
37
+ implementation, allowing for creation, reading, writing, and management of
38
+ tensor
39
+ buffers in Python.
40
+ """
41
+
42
+ def __init__(self, capsule):
43
+ """Initializes a TensorBuffer with the provided PyCapsule.
44
+
45
+ Args:
46
+ capsule: A PyCapsule containing a pointer to a LiteRtTensorBuffer.
47
+ """
48
+ self._capsule = capsule
49
+
50
+ @staticmethod
51
+ def _dtype_to_str(np_dtype):
52
+ """Converts a numpy dtype to a string representation.
53
+
54
+ Args:
55
+ np_dtype: A numpy dtype (e.g., np.float32, np.int8).
56
+
57
+ Returns:
58
+ String representation of the data type (e.g., "float32", "int8").
59
+
60
+ Raises:
61
+ ValueError: If the dtype is not supported.
62
+ """
63
+ if np_dtype == np.float32:
64
+ return "float32"
65
+ elif np_dtype == np.int32:
66
+ return "int32"
67
+ elif np_dtype == np.int8:
68
+ return "int8"
69
+ else:
70
+ raise ValueError(f"Unsupported dtype: {np_dtype}")
71
+
72
+ @classmethod
73
+ def create_from_host_memory(cls, data_array):
74
+ """Creates a new TensorBuffer referencing existing host memory.
75
+
76
+ Args:
77
+ data_array: A NumPy array (e.g., np.array([[1.0, 2.0, 3.0, 4.0]],
78
+ dtype=np.float32)). The dtype of the array is used.
79
+
80
+ Returns:
81
+ A new TensorBuffer instance.
82
+
83
+ Raises:
84
+ ValueError: If the input is not a NumPy array or has an unsupported dtype.
85
+ """
86
+ if not isinstance(data_array, np.ndarray):
87
+ raise ValueError("data_array must be a NumPy array")
88
+
89
+ dtype_str = cls._dtype_to_str(data_array.dtype)
90
+ num_elements = data_array.size
91
+
92
+ cap = _tb.CreateTensorBufferFromHostMemory(
93
+ data_array, dtype_str, num_elements
94
+ )
95
+ return cls(cap)
96
+
97
+ def write(self, data_array):
98
+ """Writes data to this tensor buffer.
99
+
100
+ Args:
101
+ data_array: A NumPy array (e.g., np.array([[1.0, 2.0, 3.0, 4.0]],
102
+ dtype=np.float32)). The dtype of the array is used.
103
+
104
+ Example:
105
+ # Using NumPy array
106
+ test_input = np.array([[1.0, 2.0, 3.0, 4.0]], dtype=np.float32)
107
+ tensor_buffer.write(test_input)
108
+
109
+ Raises:
110
+ ValueError: If the input is not a NumPy array or has an unsupported dtype.
111
+ """
112
+ if not isinstance(data_array, np.ndarray):
113
+ raise ValueError("data_array must be a NumPy array")
114
+
115
+ dtype_str = self._dtype_to_str(data_array.dtype)
116
+ _tb.WriteTensor(self._capsule, data_array.flatten().tolist(), dtype_str)
117
+
118
+ def read(self, num_elements: int, output_dtype):
119
+ """Reads data from this tensor buffer.
120
+
121
+ Args:
122
+ num_elements: Number of elements to read.
123
+ output_dtype: NumPy dtype for the output (e.g., np.float32, np.int8).
124
+
125
+ Returns:
126
+ A NumPy array containing the tensor data.
127
+
128
+ Example:
129
+ # Get output as NumPy array
130
+ output_array = tensor_buffer.read(4, np.float32).reshape((1, 4))
131
+
132
+ Raises:
133
+ ValueError: If output_dtype is not a NumPy dtype or is not supported.
134
+ """
135
+ if not isinstance(output_dtype, type) or not hasattr(
136
+ np, output_dtype.__name__
137
+ ):
138
+ raise ValueError(f"output_dtype must be a NumPy dtype (e.g., np.float32)")
139
+
140
+ dtype_str = self._dtype_to_str(output_dtype)
141
+ data_list = _tb.ReadTensor(self._capsule, num_elements, dtype_str)
142
+ return np.array(data_list, dtype=output_dtype)
143
+
144
+ def destroy(self):
145
+ """Explicitly releases resources associated with this tensor buffer.
146
+
147
+ After calling this method, the tensor buffer is no longer valid for use.
148
+ """
149
+ _tb.DestroyTensorBuffer(self._capsule)
150
+ self._capsule = None
151
+
152
+ @property
153
+ def capsule(self):
154
+ """Returns the underlying PyCapsule for direct C++ interoperability.
155
+
156
+ When the capsule is used with compiled_model methods, the ownership remains
157
+ with this TensorBuffer instance. This avoids double-free when both objects
158
+ try to destroy the same underlying tensor buffer.
159
+
160
+ Returns:
161
+ The PyCapsule containing the pointer to the LiteRtTensorBuffer, with
162
+ ownership remaining with this TensorBuffer instance.
163
+ """
164
+ # Once destroyed, shouldn't try to access the capsule again.
165
+ if self._capsule is None:
166
+ raise ValueError("TensorBuffer has been destroyed")
167
+ return self._capsule
@@ -0,0 +1,37 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: google/protobuf/timestamp.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 31,
16
+ 1,
17
+ '',
18
+ 'google/protobuf/timestamp.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ 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')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.timestamp_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _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'
35
+ _globals['_TIMESTAMP']._serialized_start=52
36
+ _globals['_TIMESTAMP']._serialized_end=95
37
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,53 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: google/protobuf/type.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 31,
16
+ 1,
17
+ '',
18
+ 'google/protobuf/type.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+ from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
26
+ from google.protobuf import source_context_pb2 as google_dot_protobuf_dot_source__context__pb2
27
+
28
+
29
+ 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')
30
+
31
+ _globals = globals()
32
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
33
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.type_pb2', _globals)
34
+ if not _descriptor._USE_C_DESCRIPTORS:
35
+ _globals['DESCRIPTOR']._loaded_options = None
36
+ _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'
37
+ _globals['_SYNTAX']._serialized_start=1447
38
+ _globals['_SYNTAX']._serialized_end=1514
39
+ _globals['_TYPE']._serialized_start=113
40
+ _globals['_TYPE']._serialized_end=345
41
+ _globals['_FIELD']._serialized_start=348
42
+ _globals['_FIELD']._serialized_end=1073
43
+ _globals['_FIELD_KIND']._serialized_start=627
44
+ _globals['_FIELD_KIND']._serialized_end=955
45
+ _globals['_FIELD_CARDINALITY']._serialized_start=957
46
+ _globals['_FIELD_CARDINALITY']._serialized_end=1073
47
+ _globals['_ENUM']._serialized_start=1076
48
+ _globals['_ENUM']._serialized_end=1299
49
+ _globals['_ENUMVALUE']._serialized_start=1301
50
+ _globals['_ENUMVALUE']._serialized_end=1384
51
+ _globals['_OPTION']._serialized_start=1386
52
+ _globals['_OPTION']._serialized_end=1445
53
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,53 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: google/protobuf/wrappers.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+ _runtime_version.ValidateProtobufRuntimeVersion(
13
+ _runtime_version.Domain.PUBLIC,
14
+ 6,
15
+ 31,
16
+ 1,
17
+ '',
18
+ 'google/protobuf/wrappers.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+
26
+
27
+ 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')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.wrappers_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _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'
35
+ _globals['_DOUBLEVALUE']._serialized_start=51
36
+ _globals['_DOUBLEVALUE']._serialized_end=79
37
+ _globals['_FLOATVALUE']._serialized_start=81
38
+ _globals['_FLOATVALUE']._serialized_end=108
39
+ _globals['_INT64VALUE']._serialized_start=110
40
+ _globals['_INT64VALUE']._serialized_end=137
41
+ _globals['_UINT64VALUE']._serialized_start=139
42
+ _globals['_UINT64VALUE']._serialized_end=167
43
+ _globals['_INT32VALUE']._serialized_start=169
44
+ _globals['_INT32VALUE']._serialized_end=196
45
+ _globals['_UINT32VALUE']._serialized_start=198
46
+ _globals['_UINT32VALUE']._serialized_end=226
47
+ _globals['_BOOLVALUE']._serialized_start=228
48
+ _globals['_BOOLVALUE']._serialized_end=254
49
+ _globals['_STRINGVALUE']._serialized_start=256
50
+ _globals['_STRINGVALUE']._serialized_end=284
51
+ _globals['_BYTESVALUE']._serialized_start=286
52
+ _globals['_BYTESVALUE']._serialized_end=313
53
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,48 @@
1
+ Metadata-Version: 2.4
2
+ Name: ai-edge-litert-nightly
3
+ Version: 2.0.4.dev20251102
4
+ Summary: LiteRT is for mobile and embedded devices.
5
+ Home-page: https://www.tensorflow.org/lite/
6
+ Author: Google AI Edge Authors
7
+ Author-email: packages@tensorflow.org
8
+ License: Apache 2.0
9
+ Keywords: litert tflite tensorflow tensor machine learning
10
+ Classifier: Development Status :: 5 - Production/Stable
11
+ Classifier: Intended Audience :: Developers
12
+ Classifier: Intended Audience :: Education
13
+ Classifier: Intended Audience :: Science/Research
14
+ Classifier: License :: OSI Approved :: Apache Software License
15
+ Classifier: Programming Language :: Python :: 3
16
+ Classifier: Programming Language :: Python :: 3.8
17
+ Classifier: Programming Language :: Python :: 3.9
18
+ Classifier: Programming Language :: Python :: 3.10
19
+ Classifier: Programming Language :: Python :: 3.11
20
+ Classifier: Topic :: Scientific/Engineering
21
+ Classifier: Topic :: Scientific/Engineering :: Mathematics
22
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
23
+ Classifier: Topic :: Software Development
24
+ Classifier: Topic :: Software Development :: Libraries
25
+ Classifier: Topic :: Software Development :: Libraries :: Python Modules
26
+ Requires-Dist: backports.strenum
27
+ Requires-Dist: flatbuffers
28
+ Requires-Dist: numpy>=1.23.2
29
+ Requires-Dist: tqdm
30
+ Requires-Dist: typing-extensions
31
+ Requires-Dist: protobuf
32
+ Provides-Extra: npu-sdk
33
+ Requires-Dist: ai-edge-litert-sdk-qualcomm~=0.1.0; extra == "npu-sdk"
34
+ Requires-Dist: ai-edge-litert-sdk-mediatek~=0.1.0; extra == "npu-sdk"
35
+ Dynamic: author
36
+ Dynamic: author-email
37
+ Dynamic: classifier
38
+ Dynamic: description
39
+ Dynamic: home-page
40
+ Dynamic: keywords
41
+ Dynamic: license
42
+ Dynamic: provides-extra
43
+ Dynamic: requires-dist
44
+ Dynamic: summary
45
+
46
+ LiteRT is the official solution for running machine learning models on mobile
47
+ and embedded devices. It enables on-device machine learning inference with low
48
+ latency and a small binary size on Android, iOS, and other operating systems.
@@ -0,0 +1,36 @@
1
+ ai_edge_litert/__init__.py,sha256=gH2YFlM35Ol-OP0iFZSszAtl5SJvwsi9xJ81QY0wqKU,34
2
+ ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=b208vF3tQRJUayfGJCiNN-h0JFAQHCkWx3mnQKGKSNY,50056
3
+ ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=R48bfhjg5-oSz8DUqDfvpztzk071ZF-snPbKHzEZMow,50104
4
+ ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=EZqmZDelToDS8vRG86Hekg2hG1pIrSMdcPtgiL6VIBg,50104
5
+ ai_edge_litert/_pywrap_modify_model_interface.so,sha256=-vhk4tGSksCJbVP48iEdO0rPQRjMs-XKxfItU-MqHwQ,50072
6
+ ai_edge_litert/_pywrap_string_util.so,sha256=Oy_WGgCX4sHhcGXI8XIeyWlw3VN1WTh4HLSNNwDJoV0,50040
7
+ ai_edge_litert/_pywrap_tensorflow_interpreter_wrapper.so,sha256=Fn-BHjXSwsiCyGxQakN7RnjDVQ7wIgjmurVfV5N1110,50104
8
+ ai_edge_litert/_pywrap_tensorflow_lite_calibration_wrapper.so,sha256=WxORp0WgNNwJ47VF4OIZ1kbtj6x2ZCEBZM_ygugST5s,50136
9
+ ai_edge_litert/_pywrap_tensorflow_lite_metrics_wrapper.so,sha256=IDQLjb0xB5hP_z2u-_-naBFhLXBlcm6DOFVWYRdgb_w,50104
10
+ ai_edge_litert/any_pb2.py,sha256=W6duyvBgx7RvePFCrJSxWagU7ddj1W9l8CsjarJJPOs,1703
11
+ ai_edge_litert/api_pb2.py,sha256=Jpn2ZFBihZFhhe47zMohsNMGIhJ9IL6VEFcLrTAlTao,2935
12
+ ai_edge_litert/compiled_model.py,sha256=xjoYYk5vHU6jP1QqjiLG5y7I5JpnBNOkl9khe6IT-Pw,7933
13
+ ai_edge_litert/descriptor_pb2.py,sha256=H0UB_-Xu4emXuqjimUeo6BaVsI0bkRex0ZWdEteBBRA,355639
14
+ ai_edge_litert/duration_pb2.py,sha256=ot6pGJwRMjEoJE0u-uoXmzqIjCLVVNb3vLSKhsf1w2c,1783
15
+ ai_edge_litert/empty_pb2.py,sha256=RtYfjWZz9WAe3fY5vDhjSo3i0M6kwZ8-kX9R_9vmyGc,1669
16
+ ai_edge_litert/field_mask_pb2.py,sha256=YVi1IasKjEQAlcIwM6HMgmW0wT6TMAXjiaiysrEY7Gc,1756
17
+ ai_edge_litert/format_converter_wrapper_pybind11.so,sha256=b_Nh1BlncXsqjzWwRNWo7Bv0sCAI8Afl7kMkOylrnxE,50088
18
+ ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
19
+ ai_edge_litert/interpreter.py,sha256=VU2Hzn60P3vqxdYAZONLoxLEIZKbHhIF8FhUshm3uKU,40707
20
+ ai_edge_litert/metrics_interface.py,sha256=dVu6SmbnQUntPgE5o6BxHVMyemwli-7F6tDfVMGrlYI,1542
21
+ ai_edge_litert/metrics_portable.py,sha256=KKvR9ZOe8j2ZeBtDo_6gWJ8kENKoOawPK3LPkevnZa8,2039
22
+ ai_edge_litert/model_runtime_info_pb2.py,sha256=iNaM8G9qxWm3LpFTacl0KP19CJxA3_N8oeDg4d-BPx8,6396
23
+ ai_edge_litert/plugin_pb2.py,sha256=dDsvFbuWV2yq4ghU6XnHMW6ZrbxYG2l8DAArIoqL8PY,3514
24
+ ai_edge_litert/profiling_info_pb2.py,sha256=RxZo_P9siirO7ktyZImOxVKMZ7g-ienzuP7rSsF61YM,3579
25
+ ai_edge_litert/pywrap_genai_ops.so,sha256=jmpe56cqSgt1rFR-4mVysQmCvSfRGzg3o1Ml6_Wt-os,50024
26
+ ai_edge_litert/schema_py_generated.py,sha256=9j4-NNxuJxeohqhYAVMgKKD9mMJsKaki1xNeguM9I6Y,676250
27
+ ai_edge_litert/source_context_pb2.py,sha256=m_GMQpv9LUK-7oKg1Vv64fTL6Bd3ZThH-fVVLtPxdqI,1775
28
+ ai_edge_litert/struct_pb2.py,sha256=DJ6P0fZe53upcah9J1jDo6BSClobouW9dHdCTItZO1A,2955
29
+ ai_edge_litert/tensor_buffer.py,sha256=3Xy7kd_aR499QrfO0YITvr_z-aU1eONga41KiDBUeew,5356
30
+ ai_edge_litert/timestamp_pb2.py,sha256=K7Gs_qOn2XAMZTNUUrJ1XcGf4OmYhdt858I_zDtZYzg,1793
31
+ ai_edge_litert/type_pb2.py,sha256=x8rTulmlk9FxtRJal7BnUsaPbn2TcLlu8D7AVoVPjMA,5065
32
+ ai_edge_litert/wrappers_pb2.py,sha256=4hQAZRGeaE5DyY6YQ7VfrqozPa_d_LCOBEYjxfQNlKs,2955
33
+ ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/METADATA,sha256=Za3I2i6mIWpEVEI7az43QlPfpLKTU-xHZqTvzVfPPYk,1935
34
+ ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/WHEEL,sha256=LPhGFfJOhbNxnN1srOnieqBgG7esDTyNtFpmf4N94K8,109
35
+ ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
36
+ ai_edge_litert_nightly-2.0.4.dev20251102.dist-info/RECORD,,
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (78.1.1)
3
+ Root-Is-Purelib: false
4
+ Tag: cp313-cp313-macosx_12_0_arm64
5
+
@@ -0,0 +1 @@
1
+ ai_edge_litert