ai-edge-litert-nightly 2.2.0.dev20260102__cp312-cp312-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.
Files changed (78) 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_interpreter_wrapper.so +0 -0
  5. ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so +0 -0
  6. ai_edge_litert/_pywrap_modify_model_interface.so +0 -0
  7. ai_edge_litert/_pywrap_string_util.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 +84 -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 +75 -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 +584 -0
  47. ai_edge_litert/internal/litertlm_core.py +58 -0
  48. ai_edge_litert/internal/litertlm_header_schema_py_generated.py +1596 -0
  49. ai_edge_litert/internal/llm_metadata_pb2.py +45 -0
  50. ai_edge_litert/internal/llm_model_type_pb2.py +51 -0
  51. ai_edge_litert/internal/sampler_params_pb2.py +39 -0
  52. ai_edge_litert/internal/token_pb2.py +38 -0
  53. ai_edge_litert/interpreter.py +1039 -0
  54. ai_edge_litert/libLiteRt.so +0 -0
  55. ai_edge_litert/libpywrap_litert_common.so +0 -0
  56. ai_edge_litert/metrics_interface.py +48 -0
  57. ai_edge_litert/metrics_portable.py +70 -0
  58. ai_edge_litert/model_runtime_info_pb2.py +66 -0
  59. ai_edge_litert/plugin_pb2.py +46 -0
  60. ai_edge_litert/profiling_info_pb2.py +47 -0
  61. ai_edge_litert/pywrap_genai_ops.so +0 -0
  62. ai_edge_litert/schema_py_generated.py +19640 -0
  63. ai_edge_litert/source_context_pb2.py +37 -0
  64. ai_edge_litert/struct_pb2.py +47 -0
  65. ai_edge_litert/tensor_buffer.py +167 -0
  66. ai_edge_litert/timestamp_pb2.py +37 -0
  67. ai_edge_litert/tools/__init__.py +0 -0
  68. ai_edge_litert/tools/apply_plugin_main +0 -0
  69. ai_edge_litert/tools/flatbuffer_utils.py +534 -0
  70. ai_edge_litert/type_pb2.py +53 -0
  71. ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so +0 -0
  72. ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so +0 -0
  73. ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so +0 -0
  74. ai_edge_litert/wrappers_pb2.py +53 -0
  75. ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/METADATA +52 -0
  76. ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/RECORD +78 -0
  77. ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/WHEEL +5 -0
  78. ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/top_level.txt +1 -0
ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/METADATA ADDED
@@ -0,0 +1,52 @@
1
+ Metadata-Version: 2.4
2
+ Name: ai-edge-litert-nightly
3
+ Version: 2.2.0.dev20260102
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
+ Provides-Extra: model-utils
36
+ Requires-Dist: lark; extra == "model-utils"
37
+ Requires-Dist: ml_dtypes; extra == "model-utils"
38
+ Requires-Dist: xdsl==0.28.0; extra == "model-utils"
39
+ Dynamic: author
40
+ Dynamic: author-email
41
+ Dynamic: classifier
42
+ Dynamic: description
43
+ Dynamic: home-page
44
+ Dynamic: keywords
45
+ Dynamic: license
46
+ Dynamic: provides-extra
47
+ Dynamic: requires-dist
48
+ Dynamic: summary
49
+
50
+ LiteRT is the official solution for running machine learning models on mobile
51
+ and embedded devices. It enables on-device machine learning inference with low
52
+ latency and a small binary size on Android, iOS, and other operating systems.
ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/RECORD ADDED
@@ -0,0 +1,78 @@
1
+ ai_edge_litert/__init__.py,sha256=WqtADn4BRrifmRJc8Id09zrpt11MfBEmbXcr7RkK8eQ,34
2
+ ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=-yAgVLhtV8rlXFqOGzAUfHBOuSKZ5SpeZfaRczRIUiw,3808
3
+ ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=tZBPPyNssub7lcsf0ikJ2c5LtAvJ5xV4jp3aVMC00os,3832
4
+ ai_edge_litert/_pywrap_litert_interpreter_wrapper.so,sha256=MO50y3mNAEGLt0eRNLKerSi9KipIW5GcKq4ci9HWUq0,3848
5
+ ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=zhUbsJbVkwnqs5WRHT5PfBnHBEPuUnzayjeSbY1l7w0,3832
6
+ ai_edge_litert/_pywrap_modify_model_interface.so,sha256=H-3teVTHOiT2cIfDv0Tzguue3b4A9HqeEmL0osSEN_E,3856
7
+ ai_edge_litert/_pywrap_string_util.so,sha256=602RhJpPy7hjlsLPvWrqUrojwiLDphWcPD1nC8XM32k,3784
8
+ ai_edge_litert/_pywrap_tensorflow_lite_calibration_wrapper.so,sha256=pBb74QBhG0QnZtaddg85SyKNMlADGlmuiTzYuxjQSXA,3896
9
+ ai_edge_litert/_pywrap_tensorflow_lite_metrics_wrapper.so,sha256=K1X5I9_2kmbRpcsPK0fj1CqhaBsQrPe082vNlSW-qtM,3872
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=s1H_tXr8BCVuJ6oRnOhIWe5fL6_L3HjZw2EVbEv2M58,3872
18
+ ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
19
+ ai_edge_litert/interpreter.py,sha256=u3QCfoVGbMVhVRu9w6FtWhyHEsyMRdBAvUK4u0ts2L4,40699
20
+ ai_edge_litert/libLiteRt.so,sha256=8em6q4blr8tfV6_PUQBYzkHtsqOiFvmMupVYCDVKyIk,7075184
21
+ ai_edge_litert/libpywrap_litert_common.so,sha256=enf8LK0hLyRzKOktJfoPwMVmLCZQYIHr-p-IlkPDX24,15362904
22
+ ai_edge_litert/metrics_interface.py,sha256=dVu6SmbnQUntPgE5o6BxHVMyemwli-7F6tDfVMGrlYI,1542
23
+ ai_edge_litert/metrics_portable.py,sha256=KKvR9ZOe8j2ZeBtDo_6gWJ8kENKoOawPK3LPkevnZa8,2039
24
+ ai_edge_litert/model_runtime_info_pb2.py,sha256=iNaM8G9qxWm3LpFTacl0KP19CJxA3_N8oeDg4d-BPx8,6396
25
+ ai_edge_litert/plugin_pb2.py,sha256=dDsvFbuWV2yq4ghU6XnHMW6ZrbxYG2l8DAArIoqL8PY,3514
26
+ ai_edge_litert/profiling_info_pb2.py,sha256=RxZo_P9siirO7ktyZImOxVKMZ7g-ienzuP7rSsF61YM,3579
27
+ ai_edge_litert/pywrap_genai_ops.so,sha256=dUQr0mUHk6L_hAlSZ6Pqvqp-p38LtK_1joCeIaKz9JI,3792
28
+ ai_edge_litert/schema_py_generated.py,sha256=9j4-NNxuJxeohqhYAVMgKKD9mMJsKaki1xNeguM9I6Y,676250
29
+ ai_edge_litert/source_context_pb2.py,sha256=m_GMQpv9LUK-7oKg1Vv64fTL6Bd3ZThH-fVVLtPxdqI,1775
30
+ ai_edge_litert/struct_pb2.py,sha256=DJ6P0fZe53upcah9J1jDo6BSClobouW9dHdCTItZO1A,2955
31
+ ai_edge_litert/tensor_buffer.py,sha256=3Xy7kd_aR499QrfO0YITvr_z-aU1eONga41KiDBUeew,5356
32
+ ai_edge_litert/timestamp_pb2.py,sha256=K7Gs_qOn2XAMZTNUUrJ1XcGf4OmYhdt858I_zDtZYzg,1793
33
+ ai_edge_litert/type_pb2.py,sha256=x8rTulmlk9FxtRJal7BnUsaPbn2TcLlu8D7AVoVPjMA,5065
34
+ ai_edge_litert/wrappers_pb2.py,sha256=4hQAZRGeaE5DyY6YQ7VfrqozPa_d_LCOBEYjxfQNlKs,2955
35
+ ai_edge_litert/aot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
36
+ ai_edge_litert/aot/aot_compile.py,sha256=CtBdFTu-a3pmPCruCZwA1FNJFLZXaSFZ_yvt8O5oBxM,5554
37
+ ai_edge_litert/aot/prepare_for_npu.py,sha256=bghtIcLXw5P6u_lHOh-oBv4u-jCoUi-kI3wF3wA1Mow,5494
38
+ ai_edge_litert/aot/ai_pack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
39
+ ai_edge_litert/aot/ai_pack/export_lib.py,sha256=-1KhA1xK6jm7bDPg9zZ7DoIbMWRfmrc1Y4PPCOqdVJU,11068
40
+ ai_edge_litert/aot/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
+ ai_edge_litert/aot/core/apply_plugin.py,sha256=xolYszNm2XDnS7VJ6XGul9UuK4ZiBvgdh0SqWwSTXx0,4598
42
+ ai_edge_litert/aot/core/common.py,sha256=E4EQVtm9EPPEhlA62AmCSj8W0LjWjCxwF5RhAnBFyvE,3138
43
+ ai_edge_litert/aot/core/components.py,sha256=DcdEDEiA6U0HT5Tt10aw_j_M1I-XtRiXTwE5hGBMEiI,2415
44
+ ai_edge_litert/aot/core/mlir_transforms.py,sha256=cW1DWZX1FDEI87Ihf3CfJ31lQmfZEBLLdptQCsK4sjM,1289
45
+ ai_edge_litert/aot/core/tflxx_util.py,sha256=fyg_K1-nVaEx207sZ88m9TOp3VcjJWpSFknv9_S8cF0,1020
46
+ ai_edge_litert/aot/core/types.py,sha256=Bvv27bUQPGQSZON2bQZ8BZTxoxLDvQki6oqtymp6fx4,10547
47
+ ai_edge_litert/aot/vendors/__init__.py,sha256=Ub-KzakcTdMnIsiI2ri0KdIpfd7wjxMhbaPKryq1Cos,1057
48
+ ai_edge_litert/aot/vendors/fallback_backend.py,sha256=LvvnXyR-EANTzh9DOYwSGstoYYcsQxtxNyc5vb9P7qE,3396
49
+ ai_edge_litert/aot/vendors/import_vendor.py,sha256=HNDkx9vdmDGtX1SupfChKdW4UtCA8Xb6xC_S7xdBKkU,3469
50
+ ai_edge_litert/aot/vendors/example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
51
+ ai_edge_litert/aot/vendors/example/example_backend.py,sha256=dgwTivJQzL1z3M_w4ixmOe_681E6cZW0bupgRPM_zOM,4182
52
+ ai_edge_litert/aot/vendors/google_tensor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
53
+ ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=dpnuzTt3LN8Ee6EVaEUXCYVcWmm_S4HII-8fhiz3VQg,5059
54
+ ai_edge_litert/aot/vendors/google_tensor/target.py,sha256=d5-pqHDWVxXWGRl7nQOTu7TukAhFsRe8NdASnyfHhds,2305
55
+ ai_edge_litert/aot/vendors/mediatek/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
56
+ ai_edge_litert/aot/vendors/mediatek/mediatek_backend.py,sha256=uFSciDUollsKciJooxn5Fo7FWdAidSG0Q0BRBxE8TQI,6074
57
+ ai_edge_litert/aot/vendors/mediatek/target.py,sha256=s-VHM9Jhodu7h9JMj5O45Nct0xtx6Pf4qS9yb2PVnqU,2274
58
+ ai_edge_litert/aot/vendors/qualcomm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
59
+ ai_edge_litert/aot/vendors/qualcomm/qualcomm_backend.py,sha256=121jVb4u1slOQ8FSf1vJBNiq1OPIMeNB9cEQrrWByJk,4864
60
+ ai_edge_litert/aot/vendors/qualcomm/target.py,sha256=Y9pdLcI7jSyy8QX4THcHMfB8KQG0bOnfqhOd0D_L8oQ,1816
61
+ ai_edge_litert/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
62
+ ai_edge_litert/internal/litertlm_builder.py,sha256=vYvxZadr-ByCyF-T7CMInc9PzvusQtXJXnA99Oa1Z3Q,19283
63
+ ai_edge_litert/internal/litertlm_core.py,sha256=xbN3CuwdCOmrVWbFJAyXpNoL33ayGeiBEa3pzu3vx6A,2022
64
+ ai_edge_litert/internal/litertlm_header_schema_py_generated.py,sha256=LiaPyX2xbGpjgU5-JLJPwKkQGO0taR2kxt5jQMj7wrM,43816
65
+ ai_edge_litert/internal/llm_metadata_pb2.py,sha256=mLp2us2_7jy9vrFkxejiQ9bj33fOczreJecrvmV6bXA,3307
66
+ ai_edge_litert/internal/llm_model_type_pb2.py,sha256=FxoKArygx1OXKcgHDLwHgt1dZSuRwF0YL5Hrp9qeEUY,5241
67
+ ai_edge_litert/internal/sampler_params_pb2.py,sha256=2sk5oGGpSre-rPEgRICwJOg7reJTuxVtwGVOjyrAriE,2069
68
+ ai_edge_litert/internal/token_pb2.py,sha256=Fliz0lpNBuxst8YW3I4_jsAEPCtSFf7KWXgn6nqIkDs,1602
69
+ ai_edge_litert/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
70
+ ai_edge_litert/tools/apply_plugin_main,sha256=r0rtyGUg0O0AxxarKUBcceWqtcVmtAhEC9k3cbIpi-Y,962464
71
+ ai_edge_litert/tools/flatbuffer_utils.py,sha256=nQqP9aZtanNEuLxIlZyXbmouB0tYF5zqlJjHPCW9MnA,17589
72
+ ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=CXzQkrO_p9zXueH4ahTqPLndrubzrsWERSm7HD3LjFM,1593528
73
+ ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=ZnuGupxM4ThyuanlhgRsCKRQkC12MROH6o4FHj5Ux7Q,529712
74
+ ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=lFcN90TqLqxbUDVkMUlrErByfoLvJGeXQe0iPmGIPqM,746240
75
+ ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/METADATA,sha256=cRCe3OKFl-9Qqfhe3p8Qz43zteqD4iSgsWYBrlWmAh0,2108
76
+ ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/WHEEL,sha256=8s5rXXO8gJtWYRoRj44BbMgEMWWy4Po1mlMXa8b8APM,113
77
+ ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
78
+ ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/RECORD,,
ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/WHEEL ADDED
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (78.1.1)
3
+ Root-Is-Purelib: false
4
+ Tag: cp312-cp312-manylinux_2_27_x86_64
5
+
ai_edge_litert_nightly-2.2.0.dev20260102.dist-info/top_level.txt ADDED
@@ -0,0 +1 @@
1
+ ai_edge_litert