ai-edge-litert-nightly 1.4.0.dev20250729__cp312-cp312-macosx_12_0_arm64.whl → 1.4.0.dev20250814__cp312-cp312-macosx_12_0_arm64.whl

ai_edge_litert/duration_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/duration.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/duration.proto'
 )

ai_edge_litert/empty_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/empty.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/empty.proto'
 )

ai_edge_litert/field_mask_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/field_mask.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/field_mask.proto'
 )

ai_edge_litert/model_runtime_info_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: tflite/profiling/proto/model_runtime_info.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'tflite/profiling/proto/model_runtime_info.proto'
 )

ai_edge_litert/plugin_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/compiler/plugin.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/compiler/plugin.proto'
 )

ai_edge_litert/profiling_info_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: tflite/profiling/proto/profiling_info.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'tflite/profiling/proto/profiling_info.proto'
 )

ai_edge_litert/source_context_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/source_context.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/source_context.proto'
 )

ai_edge_litert/struct_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/struct.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/struct.proto'
 )

ai_edge_litert/timestamp_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/timestamp.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/timestamp.proto'
 )

ai_edge_litert/type_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/type.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/type.proto'
 )

ai_edge_litert/wrappers_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: google/protobuf/wrappers.proto
-# Protobuf Python Version: 5.28.3
+# 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
@@ -11,9 +11,9 @@ from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
-    5,
-    28,
-    3,
+    6,
+    31,
+    1,
     '',
     'google/protobuf/wrappers.proto'
 )

{ai_edge_litert_nightly-1.4.0.dev20250729.dist-info → ai_edge_litert_nightly-1.4.0.dev20250814.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-edge-litert-nightly
-Version: 1.4.0.dev20250729
+Version: 1.4.0.dev20250814
 Summary: LiteRT is for mobile and embedded devices.
 Home-page: https://www.tensorflow.org/lite/
 Author: Google AI Edge Authors

ai_edge_litert_nightly-1.4.0.dev20250814.dist-info/RECORD

@@ -0,0 +1,36 @@
+ai_edge_litert/__init__.py,sha256=_8wnWQPu-Jzh5AhwdogIQzRrnYwgDBb0b51yQ0pL8ng,34
+ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=b208vF3tQRJUayfGJCiNN-h0JFAQHCkWx3mnQKGKSNY,50056
+ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=R48bfhjg5-oSz8DUqDfvpztzk071ZF-snPbKHzEZMow,50104
+ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=EZqmZDelToDS8vRG86Hekg2hG1pIrSMdcPtgiL6VIBg,50104
+ai_edge_litert/_pywrap_modify_model_interface.so,sha256=-vhk4tGSksCJbVP48iEdO0rPQRjMs-XKxfItU-MqHwQ,50072
+ai_edge_litert/_pywrap_string_util.so,sha256=Oy_WGgCX4sHhcGXI8XIeyWlw3VN1WTh4HLSNNwDJoV0,50040
+ai_edge_litert/_pywrap_tensorflow_interpreter_wrapper.so,sha256=Fn-BHjXSwsiCyGxQakN7RnjDVQ7wIgjmurVfV5N1110,50104
+ai_edge_litert/_pywrap_tensorflow_lite_calibration_wrapper.so,sha256=WxORp0WgNNwJ47VF4OIZ1kbtj6x2ZCEBZM_ygugST5s,50136
+ai_edge_litert/_pywrap_tensorflow_lite_metrics_wrapper.so,sha256=IDQLjb0xB5hP_z2u-_-naBFhLXBlcm6DOFVWYRdgb_w,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=b_Nh1BlncXsqjzWwRNWo7Bv0sCAI8Afl7kMkOylrnxE,50088
+ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
+ai_edge_litert/interpreter.py,sha256=4u9FfiaYkDxf_IxCjzeLWyTyxwG184uVxqg0fbj6tYg,40650
+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
+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=jmpe56cqSgt1rFR-4mVysQmCvSfRGzg3o1Ml6_Wt-os,50024
+ai_edge_litert/schema_py_generated.py,sha256=j1mIWi3QiiY0kVRcugNdqVcTGfRiaZmR6jb7AvwVUyM,649242
+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-1.4.0.dev20250814.dist-info/METADATA,sha256=uUlRPU3OhyxzwH9YEEuY1K8ltKCBP4j4HCL9umv5zQ8,1911
+ai_edge_litert_nightly-1.4.0.dev20250814.dist-info/WHEEL,sha256=fqm6Ds3mU75lnZFd7HRercjd2pByFTGhJZf6kGlyJBs,109
+ai_edge_litert_nightly-1.4.0.dev20250814.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-1.4.0.dev20250814.dist-info/RECORD,,

ai_edge_litert/aot/ai_pack/export_lib.py

@@ -1,281 +0,0 @@
-# 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.
-"""Utility functions for exporting models to AI pack format."""
-
-import itertools
-import os
-import pathlib
-from typing import cast
-
-from ai_edge_litert.aot.core import common
-from ai_edge_litert.aot.core import types
-from ai_edge_litert.aot.vendors import fallback_backend
-from ai_edge_litert.aot.vendors.mediatek import mediatek_backend
-from ai_edge_litert.aot.vendors.mediatek import target as mtk_target
-from ai_edge_litert.aot.vendors.qualcomm import qualcomm_backend
-from ai_edge_litert.aot.vendors.qualcomm import target as qnn_target
-
-# TODO: b/407453529 - Add unittests.
-
-
-_DEVICE_TARGETING_CONFIGURATION = """<config:device-targeting-config
-    xmlns:config="http://schemas.android.com/apk/config">
-{device_groups}
-</config:device-targeting-config>"""
-
-_DEVICE_GROUP_TEMPLATE = """    <config:device-group name="{device_group_name}">
-{device_selectors}
-    </config:device-group>"""
-
-_DEVICE_SELECTOR_TEMPLATE = """        <config:device-selector>
-            <config:system-on-chip manufacturer="{soc_man}" model="{soc_model}"/>
-        </config:device-selector>"""
-
-
-def _is_mobile_device_backend(backend: types.Backend):
-  target = backend.target
-  if backend.id() == qualcomm_backend.QualcommBackend.id():
-    target = cast(qnn_target.Target, target)
-    # Non Android QNN targets.
-    if target.soc_model in (
-        qnn_target.SocModel.SA8255,
-        qnn_target.SocModel.SA8295,
-    ):
-      return False
-  return True
-
-
-def _export_model_files_to_ai_pack(
-    compiled_models: types.CompilationResult,
-    ai_pack_dir: pathlib.Path,
-    ai_pack_name: str,
-    litert_model_name: str,
-    *,
-    separate_mtk_ai_pack: bool = True,
-):
-  """Exports the model tflite files to the AI pack directory structure.
-
-  Args:
-    compiled_models: The compiled models to export.
-    ai_pack_dir: The directory to export the AI pack to.
-    ai_pack_name: The name of the AI pack.
-    litert_model_name: The name of the model in the litert format.
-    separate_mtk_ai_pack: Whether to separate the MTK AI pack. If True, the main
-      AI pack will use the fallback model for MTK targets. The MTK AI pack will
-      contain all MTK models, and empty directories for non-MTK targets.
-  """
-  fallback_model = None
-  for backend, model in compiled_models.models_with_backend:
-    if backend.target_id == fallback_backend.FallbackBackend.id():
-      fallback_model = model
-  assert fallback_model is not None, 'Fallback model is required.'
-
-  model_export_dir = ai_pack_dir / ai_pack_name / 'src/main/assets'
-  os.makedirs(model_export_dir, exist_ok=True)
-  for backend, model in compiled_models.models_with_backend:
-    if not _is_mobile_device_backend(backend):
-      continue
-    target_id = backend.target_id
-    backend_id = backend.id()
-    if backend_id == fallback_backend.FallbackBackend.id():
-      target_id = 'other'
-    elif backend_id == mediatek_backend.MediaTekBackend.id():
-      target_id = backend.target_id.replace(
-          mtk_target.SocManufacturer.MEDIATEK, 'Mediatek'
-      )
-    group_name = 'model#group_' + target_id
-    export_dir = model_export_dir / group_name
-    os.makedirs(export_dir, exist_ok=True)
-    model_export_path = export_dir / (litert_model_name + common.DOT_TFLITE)
-    if (
-        separate_mtk_ai_pack
-        and backend_id == mediatek_backend.MediaTekBackend.id()
-    ):
-      # Use the fallback model for MTK targets in main AI pack.
-      model_to_export = fallback_model
-    else:
-      model_to_export = model
-    if not model_to_export.in_memory:
-      model_to_export.load()
-    model_to_export.save(model_export_path, export_only=True)
-
-  if separate_mtk_ai_pack:
-    _export_model_files_to_mtk_ai_pack(
-        compiled_models=compiled_models,
-        ai_pack_dir=ai_pack_dir,
-        ai_pack_name=ai_pack_name + '_mtk',
-        litert_model_name=litert_model_name + '_mtk',
-    )
-
-
-def _export_model_files_to_mtk_ai_pack(
-    compiled_models: types.CompilationResult,
-    ai_pack_dir: pathlib.Path,
-    ai_pack_name: str,
-    litert_model_name: str,
-):
-  """Exports the model tflite files to the MTK AI pack directory structure."""
-  model_export_dir = ai_pack_dir / ai_pack_name / 'src/main/assets'
-  os.makedirs(model_export_dir, exist_ok=True)
-  for backend, model in compiled_models.models_with_backend:
-    if not _is_mobile_device_backend(backend):
-      continue
-    backend_id = backend.id()
-    target_id = backend.target_id
-    if backend_id == fallback_backend.FallbackBackend.id():
-      target_id = 'other'
-    elif backend_id == mediatek_backend.MediaTekBackend.id():
-      target_id = backend.target_id.replace(
-          mtk_target.SocManufacturer.MEDIATEK, 'Mediatek'
-      )
-    group_name = 'model#group_' + target_id
-    export_dir = model_export_dir / group_name
-    os.makedirs(export_dir, exist_ok=True)
-    if backend_id != mediatek_backend.MediaTekBackend.id():
-      # Skip non-MTK targets, just create a placeholder file.
-      placeholder_file = export_dir / 'placeholder.txt'
-      placeholder_file.touch()
-      continue
-    model_export_path = export_dir / (litert_model_name + common.DOT_TFLITE)
-    if not model.in_memory:
-      model.load()
-    model.save(model_export_path, export_only=True)
-
-
-def _build_targeting_config(compiled_backends: list[types.Backend]) -> str:
-  """Builds device-targeting-config in device_targeting_configuration.xml."""
-  device_groups = []
-  for backend in compiled_backends:
-    if not _is_mobile_device_backend(backend):
-      continue
-    target = backend.target
-    device_group = _target_to_ai_pack_info(target)
-    if device_group:
-      device_groups.append(device_group)
-  device_groups = '\n'.join(device_groups)
-  return _DEVICE_TARGETING_CONFIGURATION.format(device_groups=device_groups)
-
-
-def _target_to_ai_pack_info(target: types.Target) -> str | None:
-  """Builds the device group used in device_targeting_configuration.xml."""
-  if isinstance(target, qnn_target.Target):
-    group_name = str(target)
-    selector = _process_qnn_target(target)
-    device_selectors = [
-        _DEVICE_SELECTOR_TEMPLATE.format(soc_man=man, soc_model=model)
-        for man, model in selector
-    ]
-    device_selectors = '\n'.join(device_selectors)
-    device_group = _DEVICE_GROUP_TEMPLATE.format(
-        device_group_name=group_name, device_selectors=device_selectors
-    )
-    return device_group
-  elif isinstance(target, mtk_target.Target):
-    group_name = str(target).replace(
-        mtk_target.SocManufacturer.MEDIATEK, 'Mediatek'
-    )
-    # TODO: b/407453529 - Support MTK SDK Version / OS version in selector.
-    selector = _process_mtk_target(target)
-    device_selector = _DEVICE_SELECTOR_TEMPLATE.format(
-        soc_man=selector[0], soc_model=selector[1]
-    )
-    device_group = _DEVICE_GROUP_TEMPLATE.format(
-        device_group_name=group_name, device_selectors=device_selector
-    )
-    return device_group
-  elif isinstance(target, fallback_backend.FallbackTarget):
-    # Don't need to have device selector for fallback target.
-    return None
-  else:
-    print('unsupported target ', target)
-    return None
-
-
-# TODO: b/407453529 - Auto-generate this function from CSVs.
-def _process_qnn_target(target: qnn_target.Target) -> list[tuple[str, str]]:
-  """Returns the list of (manufacturer, model) for the given QNN target."""
-  # Play cannot distinguish between Qualcomm and QTI for now.
-  manufacturer = ['Qualcomm', 'QTI']
-  models = [str(target.soc_model)]
-  return list(itertools.product(manufacturer, models))
-
-
-# TODO: b/407453529 - Auto-generate this function from CSVs.
-def _process_mtk_target(
-    target: mtk_target.Target,
-) -> tuple[str, str]:
-  """Returns tuple of (manufacturer, model) for the given MTK target."""
-  # Play cannot distinguish between Qualcomm and QTI for now.
-  return str(target.soc_manufacturer).replace(
-      mtk_target.SocManufacturer.MEDIATEK, 'Mediatek'
-  ), str(target.soc_model)
-
-
-def _write_targeting_config(
-    compiled_models: types.CompilationResult, ai_pack_dir: pathlib.Path
-) -> None:
-  """Writes device_targeting_configuration.xml for the given compiled models."""
-  compiled_backends = [x for x, _ in compiled_models.models_with_backend]
-  targeting_config = _build_targeting_config(
-      compiled_backends=compiled_backends
-  )
-
-  targeting_config_path = ai_pack_dir / 'device_targeting_configuration.xml'
-  targeting_config_path.write_text(targeting_config)
-
-
-def export(
-    compiled_models: types.CompilationResult,
-    ai_pack_dir: pathlib.Path | str,
-    ai_pack_name: str,
-    litert_model_name: str,
-) -> None:
-  """Exports the compiled models to AI pack format.
-
-  This function will export the compiled models to corresponding directory
-  structure:
-
-  {ai_pack_dir}/
-    AiPackManifest.xml
-    device_targeting_configuration.xml
-    {ai_pack_name}/src/main/assets/
-      model#group_target_1/
-        {litert_model_name}.tflite
-      model#group_target_2/
-        {litert_model_name}.tflite
-      model#group_target_3/
-        {litert_model_name}.tflite
-      model#group_other/
-        {litert_model_name}.tflite
-
-  Args:
-    compiled_models: The compiled models to export.
-    ai_pack_dir: The directory to export the AI pack to.
-    ai_pack_name: The name of the AI pack.
-    litert_model_name: The name of the model in the litert format.
-  """
-  if isinstance(ai_pack_dir, str):
-    ai_pack_dir = pathlib.Path(ai_pack_dir)
-
-  ai_pack_dir.mkdir(parents=True, exist_ok=True)
-
-  _export_model_files_to_ai_pack(
-      compiled_models=compiled_models,
-      ai_pack_dir=ai_pack_dir,
-      ai_pack_name=ai_pack_name,
-      litert_model_name=litert_model_name,
-  )
-  _write_targeting_config(
-      compiled_models=compiled_models, ai_pack_dir=ai_pack_dir
-  )