@@ -1415,6 +1415,26 @@ def patch_workspace(path: Path) -> None:
14151415"""
14161416
14171417
1418+ PORTABLE_KERNELS_QUEUE_DEPS_PATCH = """--- a/tensorflow/core/kernels/BUILD
1419+ +++ b/tensorflow/core/kernels/BUILD
1420+ @@ -6947,13 +6947,14 @@
1421+ ],
1422+ visibility = ["//visibility:public"],
1423+ deps = [
1424+ "//tensorflow/core:portable_tensorflow_lib_lite",
1425+ "//tensorflow/core:protos_all_cc_impl",
1426+ + ":padding_fifo_queue",
1427+ "//third_party/eigen3",
1428+ "//third_party/fft2d:fft2d_headers",
1429+ "@com_google_absl//absl/base",
1430+ "@com_google_protobuf//:protobuf",
1431+ "@fft2d",
1432+ "@gemmlowp",
1433+ ],
1434+ alwayslink = 1,
1435+ """
1436+
1437+
14181438HUNK_HEADER_RE = re .compile (
14191439 r"^@@ -(?P<old_start>\d+)(?:,(?P<old_count>\d+))? "
14201440 r"\+(?P<new_start>\d+)(?:,(?P<new_count>\d+))? @@(?P<section>.*)$"
@@ -1501,6 +1521,7 @@ def write_tensorflow_android_absl_patch(path: Path) -> None:
15011521 text += SAVED_MODEL_ANDROID_LOADER_PATCH
15021522 text += TENSORFLOW_FRAMEWORK_ANDROID_PATCH
15031523 text += normalize_unified_diff_hunk_counts (ANDROID_PORTABLE_LIB_SHIM_PATCH )
1524+ text += PORTABLE_KERNELS_QUEUE_DEPS_PATCH
15041525 if not text .endswith ("\n " ):
15051526 text += "\n "
15061527 path .write_text (text , encoding = "utf-8" )
0 commit comments