build: shim android portable tensorflow libs

Author: Apricityx

scripts/patch_tfjava.py

@@ -903,6 +903,85 @@ def patch_workspace(path: Path) -> None:
 """
 
 
+ANDROID_PORTABLE_LIB_SHIM_PATCH = """--- a/tensorflow/core/BUILD
++++ b/tensorflow/core/BUILD
+@@ -1378,7 +1378,10 @@
+ cc_library(
+     name = "portable_tensorflow_lib_lite",
+-    srcs = if_mobile([":mobile_srcs"]),
++    srcs = select({
++        "//tensorflow:android": [],
++        "//conditions:default": if_mobile([":mobile_srcs"]),
++    }),
+     copts = tf_copts(android_optimization_level_override = None) + tf_opts_nortti_if_lite_protos() + if_ios(["-Os"]),
+     defines = ["SUPPORT_SELECTIVE_REGISTRATION"] + tf_portable_full_lite_protos(
+         full = [],
+@@ -1390,11 +1393,22 @@
+         "notap",
+     ],
+     visibility = ["//visibility:public"],
+-    deps = [
+-        ":protos_all_cc_impl",
+-        "//tensorflow/core/util:stats_calculator_portable",
+-        "//tensorflow/core:mobile_additional_lib_deps",
+-    ] + tf_portable_deps_no_runtime(),
++    deps = select({
++        "//tensorflow:android": [
++            ":core_cpu",
++            ":core_cpu_internal",
++            ":framework",
++            ":framework_internal",
++            ":lib",
++            ":lib_internal",
++            ":protos_all_cc",
++        ],
++        "//conditions:default": [
++            ":protos_all_cc_impl",
++            "//tensorflow/core/util:stats_calculator_portable",
++            "//tensorflow/core:mobile_additional_lib_deps",
++        ] + tf_portable_deps_no_runtime(),
++    }),
+     alwayslink = 1,
+ )
+@@ -1485,7 +1499,10 @@
+ cc_library(
+     name = "portable_tensorflow_lib",
+-    srcs = if_mobile([":portable_op_registrations_and_gradients"]),
++    srcs = select({
++        "//tensorflow:android": [],
++        "//conditions:default": if_mobile([":portable_op_registrations_and_gradients"]),
++    }),
+     copts = tf_copts() + tf_opts_nortti_if_lite_protos(),
+     features = tf_features_nomodules_if_mobile(),
+     tags = [
+@@ -1494,12 +1511,16 @@
+         "notap",
+     ],
+     visibility = ["//visibility:public"],
+-    deps = [
+-        ":portable_tensorflow_lib_lite",
+-        ":protos_all_cc_impl",
+-        "//tensorflow/core/kernels:portable_tensorflow_kernels",
+-        "//third_party/eigen3",
+-        "@com_google_protobuf//:protobuf",
+-    ],
++    deps = select({
++        "//tensorflow:android": [
++            ":portable_tensorflow_lib_lite",
++        ],
++        "//conditions:default": [
++            ":portable_tensorflow_lib_lite",
++            ":protos_all_cc_impl",
++            "//tensorflow/core/kernels:portable_tensorflow_kernels",
++            "//third_party/eigen3",
++            "@com_google_protobuf//:protobuf",
++        ],
++    }),
+     alwayslink = 1,
+ )
+"""
+
+
 def write_tensorflow_android_absl_patch(path: Path) -> None:
     path.parent.mkdir(parents=True, exist_ok=True)
     absl_patch_text = (
@@ -929,6 +1008,7 @@ def write_tensorflow_android_absl_patch(path: Path) -> None:
     text += TF_C_API_EXPERIMENTAL_CC_PATCH
     text += SAVED_MODEL_ANDROID_LOADER_PATCH
     text += TENSORFLOW_FRAMEWORK_ANDROID_PATCH
+    text += ANDROID_PORTABLE_LIB_SHIM_PATCH
     if not text.endswith("\n"):
         text += "\n"
     path.write_text(text, encoding="utf-8")