Use pyrwap rules for LiteRT repo to fix duplicate registration errors

PiperOrigin-RevId: 758995263

Author: ecalubaquib

tflite/python/BUILD

@@ -1,7 +1,8 @@
 load("@flatbuffers//:build_defs.bzl", "flatbuffer_py_library")
 load("@org_tensorflow//tensorflow:pytype.default.bzl", "pytype_strict_contrib_test", "pytype_strict_library")
 load("@org_tensorflow//tensorflow:strict.default.bzl", "py_strict_binary", "py_strict_library", "py_strict_test")
-load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
+load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable", "pywrap_library")
+load("@org_tensorflow//tensorflow/core/platform:build_config_root.bzl", "if_pywrap")
 load("@rules_shell//shell:sh_test.bzl", "sh_test")
 load("//tflite:special_rules.bzl", "internal_visibility_allowlist")
 
@@ -474,7 +475,9 @@ pytype_strict_library(
         "@org_tensorflow//tensorflow/python/framework:tensor_shape",
         "@org_tensorflow//tensorflow/python/util:deprecation",
         "@org_tensorflow//tensorflow/python/util:tf_export",
-    ],
+    ] + if_pywrap(
+        if_true = [":pywrap_tflite"],
+    ),
 )
 
 py_strict_library(
@@ -515,7 +518,9 @@ pytype_strict_contrib_test(
         "@org_tensorflow//tensorflow/python/ops:math_ops",
         "@org_tensorflow//tensorflow/python/platform:client_testlib",
         "@org_tensorflow//third_party/py/numpy",
-    ],
+    ] + if_pywrap(
+        if_true = [":pywrap_tflite"],
+    ),
 )
 
 py_strict_library(
@@ -633,3 +638,21 @@ py_strict_test(
         "@org_tensorflow//tensorflow/python/trackable:autotrackable",
     ],
 )
+
+pywrap_library(
+    name = "pywrap_tflite",
+    common_lib_versions = {
+        "framework": "2",
+    },
+    pywrap_count = 7,
+    visibility = ["//visibility:public"],
+    deps = [
+        "//tflite/python/analyzer_wrapper:_pywrap_analyzer_wrapper",
+        "//tflite/python/interpreter_wrapper:_pywrap_tensorflow_interpreter_wrapper",
+        "//tflite/python/metrics:_pywrap_tensorflow_lite_metrics_wrapper",
+        "//tflite/python/optimize:_pywrap_tensorflow_lite_calibration_wrapper",
+        "//tflite/testing:_pywrap_string_util",
+        "//tflite/tools/optimize/python:_pywrap_modify_model_interface",
+        "//tflite/tools/optimize/sparsity:format_converter_wrapper_pybind11",
+    ],
+)

tflite/python/metrics/BUILD

@@ -36,7 +36,7 @@ pybind_extension(
     pytype_srcs = [
         "_pywrap_tensorflow_lite_metrics_wrapper.pyi",
     ],
-    visibility = ["__subpackages__"],
+    visibility = ["//visibility:public"],
     deps = [
         ":metrics_wrapper_lib",
         "@com_google_protobuf//:protobuf",

tflite/tools/optimize/python/BUILD

@@ -54,6 +54,11 @@ py_strict_library(
 pybind_extension(
     name = "_pywrap_modify_model_interface",
     srcs = ["modify_model_interface.cc"],
+    # copybara:comment_begin(oss-only)
+    common_lib_packages = [
+        "tflite/python",
+    ],
+    # copybara:comment_end
     enable_stub_generation = True,
     pytype_srcs = [
         "_pywrap_modify_model_interface.pyi",

tflite/tools/optimize/sparsity/BUILD

@@ -1,5 +1,6 @@
 load("@org_tensorflow//tensorflow:strict.default.bzl", "py_strict_test")
 load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "pybind_extension")
+load("@org_tensorflow//tensorflow/core/platform:build_config_root.bzl", "if_pywrap")
 
 package(
     # copybara:uncomment default_applicable_licenses = ["@org_tensorflow//tensorflow:license"],
@@ -12,6 +13,11 @@ package(
 pybind_extension(
     name = "format_converter_wrapper_pybind11",
     srcs = ["format_converter_wrapper_pybind11.cc"],
+    # copybara:comment_begin(oss-only)
+    common_lib_packages = [
+        "tflite/python",
+    ],
+    # copybara:comment_end
     copts = [
         "-fexceptions",
         "-fno-strict-aliasing",
@@ -37,5 +43,7 @@ py_strict_test(
         "@absl_py//absl/testing:absltest",
         #internal proto upb dep
         "@org_tensorflow//third_party/py/numpy",
-    ],
+    ] + if_pywrap(
+        if_true = ["//tflite/python:pywrap_tflite"],
+    ),
 )