ai-edge-litert-nightly 2.0.4.dev20250928__cp312-cp312-manylinux_2_27_x86_64.whl → 2.0.4.dev20250929__cp312-cp312-manylinux_2_27_x86_64.whl

ai_edge_litert/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.4.dev20250928"
+__version__ = "2.0.4.dev20250929"

ai_edge_litert/aot/vendors/__init__.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 # ==============================================================================
 """Vendor backends for LiteRt."""
+import os
 
-from ai_edge_litert.aot.vendors.google_tensor import google_tensor_backend as _
 from ai_edge_litert.aot.vendors.mediatek import mediatek_backend as _
 from ai_edge_litert.aot.vendors.qualcomm import qualcomm_backend as _
+
+if os.environ.get("GOOGLE_TENSOR_COMPILER_LIB") is not None:
+  from ai_edge_litert.aot.vendors.google_tensor import google_tensor_backend as _  # pylint: disable=g-import-not-at-top

ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py

@@ -120,7 +120,7 @@ def _apply_plugin(
     # Otherwise we use the default library path.
     plugin_path = common.get_resource(COMPILER_PLUGIN_LIB_PATH)
     lib_dir = os.path.dirname(plugin_path)
-    sdk_libs_dir = os.environ.get("LITERT_GOOGLE_TENSOR_SDK", None)
+    sdk_libs_dir = os.environ.get("GOOGLE_TENSOR_COMPILER_LIB", None)
 
     extra_kwargs = {"libs": lib_dir, "sdk_libs_path": sdk_libs_dir}
   except FileNotFoundError:

ai_edge_litert/interpreter.py

@@ -664,6 +664,7 @@ class Interpreter:
             'scales': tensor_quantization_params[0],
             'zero_points': tensor_quantization_params[1],
             'quantized_dimension': tensor_quantization_params[2],
+            'block_size': tensor_quantization_params[3],
         },
         'sparsity_parameters': tensor_sparsity_params,
     }

{ai_edge_litert_nightly-2.0.4.dev20250928.dist-info → ai_edge_litert_nightly-2.0.4.dev20250929.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-edge-litert-nightly
-Version: 2.0.4.dev20250928
+Version: 2.0.4.dev20250929
 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.dev20250928.dist-info → ai_edge_litert_nightly-2.0.4.dev20250929.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ai_edge_litert/__init__.py,sha256=VXg4iVzo7FpE4hWfgbsSJQdhE82AnyRSC3Yf-kx9MTg,34
+ai_edge_litert/__init__.py,sha256=Cyb1N1NWawJcf2c1DlOoLuqeWhaA2J4Cxt9pB8sJ24w,34
 ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=-yAgVLhtV8rlXFqOGzAUfHBOuSKZ5SpeZfaRczRIUiw,3808
 ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=tZBPPyNssub7lcsf0ikJ2c5LtAvJ5xV4jp3aVMC00os,3832
 ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=zhUbsJbVkwnqs5WRHT5PfBnHBEPuUnzayjeSbY1l7w0,3832
@@ -16,9 +16,9 @@ ai_edge_litert/empty_pb2.py,sha256=RtYfjWZz9WAe3fY5vDhjSo3i0M6kwZ8-kX9R_9vmyGc,1
 ai_edge_litert/field_mask_pb2.py,sha256=YVi1IasKjEQAlcIwM6HMgmW0wT6TMAXjiaiysrEY7Gc,1756
 ai_edge_litert/format_converter_wrapper_pybind11.so,sha256=s1H_tXr8BCVuJ6oRnOhIWe5fL6_L3HjZw2EVbEv2M58,3872
 ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
-ai_edge_litert/interpreter.py,sha256=4u9FfiaYkDxf_IxCjzeLWyTyxwG184uVxqg0fbj6tYg,40650
-ai_edge_litert/libLiteRtRuntimeCApi.so,sha256=vk-NBzl1Ss1QFhth-RSnA8LrmInQcy7D6NPxXl4L0kE,6815664
-ai_edge_litert/libpywrap_litert_common.so,sha256=ji6NuW7k1i58tO7FX-tgMl75R-WkDrIU_gLkZP0YZBM,15076888
+ai_edge_litert/interpreter.py,sha256=VU2Hzn60P3vqxdYAZONLoxLEIZKbHhIF8FhUshm3uKU,40707
+ai_edge_litert/libLiteRtRuntimeCApi.so,sha256=c9F3ySOVbnFJf3X8OlohoUyxnZIKx0K7Nhzk9Y_UOZc,6819216
+ai_edge_litert/libpywrap_litert_common.so,sha256=D8S9RMQl_iEIA_MlEWrTV3ZGpisF2gUhL5G0eW35-ow,15077488
 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
@@ -44,13 +44,13 @@ ai_edge_litert/aot/core/components.py,sha256=DcdEDEiA6U0HT5Tt10aw_j_M1I-XtRiXTwE
 ai_edge_litert/aot/core/mlir_transforms.py,sha256=cW1DWZX1FDEI87Ihf3CfJ31lQmfZEBLLdptQCsK4sjM,1289
 ai_edge_litert/aot/core/tflxx_util.py,sha256=fyg_K1-nVaEx207sZ88m9TOp3VcjJWpSFknv9_S8cF0,1020
 ai_edge_litert/aot/core/types.py,sha256=Bvv27bUQPGQSZON2bQZ8BZTxoxLDvQki6oqtymp6fx4,10547
-ai_edge_litert/aot/vendors/__init__.py,sha256=OdqpkjR-n0_9sRN7a0cW3Ih0j5S6y994iJRGRkmkcMo,944
+ai_edge_litert/aot/vendors/__init__.py,sha256=Ub-KzakcTdMnIsiI2ri0KdIpfd7wjxMhbaPKryq1Cos,1057
 ai_edge_litert/aot/vendors/fallback_backend.py,sha256=LvvnXyR-EANTzh9DOYwSGstoYYcsQxtxNyc5vb9P7qE,3396
 ai_edge_litert/aot/vendors/import_vendor.py,sha256=HNDkx9vdmDGtX1SupfChKdW4UtCA8Xb6xC_S7xdBKkU,3469
 ai_edge_litert/aot/vendors/example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/vendors/example/example_backend.py,sha256=dgwTivJQzL1z3M_w4ixmOe_681E6cZW0bupgRPM_zOM,4182
 ai_edge_litert/aot/vendors/google_tensor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=xWDwsw1gjb7uPoXyoaiI5cZ6uKCRK87VrVLyjvPyrcc,5057
+ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=dpnuzTt3LN8Ee6EVaEUXCYVcWmm_S4HII-8fhiz3VQg,5059
 ai_edge_litert/aot/vendors/google_tensor/target.py,sha256=jzJVgKr2grhaazs-Wo-yHLb9QjGWDENA44uAjO7s4p8,2279
 ai_edge_litert/aot/vendors/mediatek/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/vendors/mediatek/mediatek_backend.py,sha256=uFSciDUollsKciJooxn5Fo7FWdAidSG0Q0BRBxE8TQI,6074
@@ -64,11 +64,11 @@ ai_edge_litert/internal/litertlm_core.py,sha256=twP4-6kfqmOaqyPnJ5np8CMEn8loVmk-
 ai_edge_litert/internal/litertlm_header_schema_py_generated.py,sha256=xnijnbk5eipnsDwr2pDc8qfLG_k5S3o9UlhvVawnypg,42860
 ai_edge_litert/internal/llm_metadata_pb2.py,sha256=ABbevvRn5VDBDG39zq-rsvH21qdO6OoK_ahjPqjXLZo,3028
 ai_edge_litert/internal/sampler_params_pb2.py,sha256=2sk5oGGpSre-rPEgRICwJOg7reJTuxVtwGVOjyrAriE,2069
-ai_edge_litert/tools/apply_plugin_main,sha256=cLHcBIPX2nHkPvHKkyqLtRapywMxBNlciYrb2jzrUrs,3636232
-ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=Lbb6h_2a56s50ik-S8z64swbu-DweNdkSQJ2obqP_J4,227472
+ai_edge_litert/tools/apply_plugin_main,sha256=C5mAD5t7VQSIJg5eOOakS9FHY_KSE0C-kWhcrzE8RHY,3637232
+ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=iRrKZHBp91kiQrFKdmt9UJEMZn7ehoHyPyXLIRLXWnc,227520
 ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=DMLZ1TaYEkdwV-jf7QxqbvDT_iIRlnyOjRtjB_8rdHw,508856
-ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=I1OWzUDwph_9a8LpSN3_znWd0qh4KJd_d-2_aJbYuLo,629776
-ai_edge_litert_nightly-2.0.4.dev20250928.dist-info/METADATA,sha256=UDlhIJo54OyCYeV2deIoCXdCecFza-6qgkxUpvX1UGk,1935
-ai_edge_litert_nightly-2.0.4.dev20250928.dist-info/WHEEL,sha256=8s5rXXO8gJtWYRoRj44BbMgEMWWy4Po1mlMXa8b8APM,113
-ai_edge_litert_nightly-2.0.4.dev20250928.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
-ai_edge_litert_nightly-2.0.4.dev20250928.dist-info/RECORD,,
+ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=cMEI8vym4kt49IasFWzsQpx43tiT6uATV_U4QBKU-H4,635640
+ai_edge_litert_nightly-2.0.4.dev20250929.dist-info/METADATA,sha256=lou_wAvJci9okqHN3HMXZLxx2zvmZdzJrGxoQjd4sHM,1935
+ai_edge_litert_nightly-2.0.4.dev20250929.dist-info/WHEEL,sha256=8s5rXXO8gJtWYRoRj44BbMgEMWWy4Po1mlMXa8b8APM,113
+ai_edge_litert_nightly-2.0.4.dev20250929.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-2.0.4.dev20250929.dist-info/RECORD,,