ai-edge-litert-nightly 2.0.4.dev20251030__cp313-cp313-manylinux_2_27_x86_64.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 (74) 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/aot/__init__.py +0 -0
  12. ai_edge_litert/aot/ai_pack/__init__.py +0 -0
  13. ai_edge_litert/aot/ai_pack/export_lib.py +300 -0
  14. ai_edge_litert/aot/aot_compile.py +153 -0
  15. ai_edge_litert/aot/core/__init__.py +0 -0
  16. ai_edge_litert/aot/core/apply_plugin.py +148 -0
  17. ai_edge_litert/aot/core/common.py +97 -0
  18. ai_edge_litert/aot/core/components.py +93 -0
  19. ai_edge_litert/aot/core/mlir_transforms.py +36 -0
  20. ai_edge_litert/aot/core/tflxx_util.py +30 -0
  21. ai_edge_litert/aot/core/types.py +374 -0
  22. ai_edge_litert/aot/prepare_for_npu.py +152 -0
  23. ai_edge_litert/aot/vendors/__init__.py +22 -0
  24. ai_edge_litert/aot/vendors/example/__init__.py +0 -0
  25. ai_edge_litert/aot/vendors/example/example_backend.py +157 -0
  26. ai_edge_litert/aot/vendors/fallback_backend.py +128 -0
  27. ai_edge_litert/aot/vendors/google_tensor/__init__.py +0 -0
  28. ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py +168 -0
  29. ai_edge_litert/aot/vendors/google_tensor/target.py +83 -0
  30. ai_edge_litert/aot/vendors/import_vendor.py +132 -0
  31. ai_edge_litert/aot/vendors/mediatek/__init__.py +0 -0
  32. ai_edge_litert/aot/vendors/mediatek/mediatek_backend.py +196 -0
  33. ai_edge_litert/aot/vendors/mediatek/target.py +94 -0
  34. ai_edge_litert/aot/vendors/qualcomm/__init__.py +0 -0
  35. ai_edge_litert/aot/vendors/qualcomm/qualcomm_backend.py +161 -0
  36. ai_edge_litert/aot/vendors/qualcomm/target.py +74 -0
  37. ai_edge_litert/api_pb2.py +43 -0
  38. ai_edge_litert/compiled_model.py +250 -0
  39. ai_edge_litert/descriptor_pb2.py +3361 -0
  40. ai_edge_litert/duration_pb2.py +37 -0
  41. ai_edge_litert/empty_pb2.py +37 -0
  42. ai_edge_litert/field_mask_pb2.py +37 -0
  43. ai_edge_litert/format_converter_wrapper_pybind11.so +0 -0
  44. ai_edge_litert/hardware_accelerator.py +22 -0
  45. ai_edge_litert/internal/__init__.py +0 -0
  46. ai_edge_litert/internal/litertlm_builder.py +573 -0
  47. ai_edge_litert/internal/litertlm_core.py +58 -0
  48. ai_edge_litert/internal/litertlm_header_schema_py_generated.py +1595 -0
  49. ai_edge_litert/internal/llm_metadata_pb2.py +45 -0
  50. ai_edge_litert/internal/sampler_params_pb2.py +39 -0
  51. ai_edge_litert/interpreter.py +1039 -0
  52. ai_edge_litert/libLiteRtRuntimeCApi.so +0 -0
  53. ai_edge_litert/libpywrap_litert_common.so +0 -0
  54. ai_edge_litert/metrics_interface.py +48 -0
  55. ai_edge_litert/metrics_portable.py +70 -0
  56. ai_edge_litert/model_runtime_info_pb2.py +66 -0
  57. ai_edge_litert/plugin_pb2.py +46 -0
  58. ai_edge_litert/profiling_info_pb2.py +47 -0
  59. ai_edge_litert/pywrap_genai_ops.so +0 -0
  60. ai_edge_litert/schema_py_generated.py +19304 -0
  61. ai_edge_litert/source_context_pb2.py +37 -0
  62. ai_edge_litert/struct_pb2.py +47 -0
  63. ai_edge_litert/tensor_buffer.py +167 -0
  64. ai_edge_litert/timestamp_pb2.py +37 -0
  65. ai_edge_litert/tools/apply_plugin_main +0 -0
  66. ai_edge_litert/type_pb2.py +53 -0
  67. ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so +0 -0
  68. ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so +0 -0
  69. ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so +0 -0
  70. ai_edge_litert/wrappers_pb2.py +53 -0
  71. ai_edge_litert_nightly-2.0.4.dev20251030.dist-info/METADATA +48 -0
  72. ai_edge_litert_nightly-2.0.4.dev20251030.dist-info/RECORD +74 -0
  73. ai_edge_litert_nightly-2.0.4.dev20251030.dist-info/WHEEL +5 -0
  74. ai_edge_litert_nightly-2.0.4.dev20251030.dist-info/top_level.txt +1 -0
ai_edge_litert/internal/llm_metadata_pb2.py ADDED
@@ -0,0 +1,45 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: litert/python/internal/llm_metadata.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
+ 'litert/python/internal/llm_metadata.proto'
19
+ )
20
+ # @@protoc_insertion_point(imports)
21
+
22
+ _sym_db = _symbol_database.Default()
23
+
24
+
25
+ from ai_edge_litert.internal import sampler_params_pb2 as litert_dot_python_dot_internal_dot_sampler__params__pb2
26
+
27
+
28
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)litert/python/internal/llm_metadata.proto\x12\x0flitert.lm.proto\x1a+litert/python/internal/sampler_params.proto\"/\n\rPromptAffixes\x12\x0e\n\x06prefix\x18\x01 \x01(\t\x12\x0e\n\x06suffix\x18\x02 \x01(\t\"\xcb\x01\n\x0fPromptTemplates\x12\x31\n\x04user\x18\x01 \x01(\x0b\x32\x1e.litert.lm.proto.PromptAffixesH\x00\x88\x01\x01\x12\x32\n\x05model\x18\x02 \x01(\x0b\x32\x1e.litert.lm.proto.PromptAffixesH\x01\x88\x01\x01\x12\x33\n\x06system\x18\x03 \x01(\x0b\x32\x1e.litert.lm.proto.PromptAffixesH\x02\x88\x01\x01\x42\x07\n\x05_userB\x08\n\x06_modelB\t\n\x07_system\"\xb3\x02\n\x0bLlmMetadata\x12\x30\n\x0bstart_token\x18\x01 \x01(\x0b\x32\x1b.litert.lm.proto.TokenUnion\x12\x30\n\x0bstop_tokens\x18\x02 \x03(\x0b\x32\x1b.litert.lm.proto.TokenUnion\x12?\n\x10prompt_templates\x18\x03 \x01(\x0b\x32 .litert.lm.proto.PromptTemplatesH\x00\x88\x01\x01\x12?\n\x0esampler_params\x18\x04 \x01(\x0b\x32\".litert.lm.proto.SamplerParametersH\x01\x88\x01\x01\x12\x16\n\x0emax_num_tokens\x18\x05 \x01(\x05\x42\x13\n\x11_prompt_templatesB\x11\n\x0f_sampler_params\"`\n\nTokenUnion\x12.\n\ttoken_ids\x18\x01 \x01(\x0b\x32\x19.litert.lm.proto.TokenIdsH\x00\x12\x13\n\ttoken_str\x18\x02 \x01(\tH\x00\x42\r\n\x0btoken_union\"\x17\n\x08TokenIds\x12\x0b\n\x03ids\x18\x01 \x03(\x05\x62\x06proto3')
29
+
30
+ _globals = globals()
31
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
32
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'litert.python.internal.llm_metadata_pb2', _globals)
33
+ if not _descriptor._USE_C_DESCRIPTORS:
34
+ DESCRIPTOR._loaded_options = None
35
+ _globals['_PROMPTAFFIXES']._serialized_start=107
36
+ _globals['_PROMPTAFFIXES']._serialized_end=154
37
+ _globals['_PROMPTTEMPLATES']._serialized_start=157
38
+ _globals['_PROMPTTEMPLATES']._serialized_end=360
39
+ _globals['_LLMMETADATA']._serialized_start=363
40
+ _globals['_LLMMETADATA']._serialized_end=670
41
+ _globals['_TOKENUNION']._serialized_start=672
42
+ _globals['_TOKENUNION']._serialized_end=768
43
+ _globals['_TOKENIDS']._serialized_start=770
44
+ _globals['_TOKENIDS']._serialized_end=793
45
+ # @@protoc_insertion_point(module_scope)
ai_edge_litert/internal/sampler_params_pb2.py ADDED
@@ -0,0 +1,39 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: litert/python/internal/sampler_params.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
+ 'litert/python/internal/sampler_params.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+litert/python/internal/sampler_params.proto\x12\x0flitert.lm.proto\"\xd1\x01\n\x11SamplerParameters\x12\x35\n\x04type\x18\x01 \x01(\x0e\x32\'.litert.lm.proto.SamplerParameters.Type\x12\t\n\x01k\x18\x02 \x01(\x05\x12\t\n\x01p\x18\x03 \x01(\x02\x12\x13\n\x0btemperature\x18\x04 \x01(\x02\x12\x11\n\x04seed\x18\x05 \x01(\x05H\x00\x88\x01\x01\">\n\x04Type\x12\x14\n\x10TYPE_UNSPECIFIED\x10\x00\x12\t\n\x05TOP_K\x10\x01\x12\t\n\x05TOP_P\x10\x02\x12\n\n\x06GREEDY\x10\x03\x42\x07\n\x05_seedB4\n\x1a\x63om.google.litert.lm.protoB\x16SamplerParametersProtob\x06proto3')
28
+
29
+ _globals = globals()
30
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
31
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'litert.python.internal.sampler_params_pb2', _globals)
32
+ if not _descriptor._USE_C_DESCRIPTORS:
33
+ _globals['DESCRIPTOR']._loaded_options = None
34
+ _globals['DESCRIPTOR']._serialized_options = b'\n\032com.google.litert.lm.protoB\026SamplerParametersProto'
35
+ _globals['_SAMPLERPARAMETERS']._serialized_start=65
36
+ _globals['_SAMPLERPARAMETERS']._serialized_end=274
37
+ _globals['_SAMPLERPARAMETERS_TYPE']._serialized_start=203
38
+ _globals['_SAMPLERPARAMETERS_TYPE']._serialized_end=265
39
+ # @@protoc_insertion_point(module_scope)