Skip to content

Commit 5ec1920

Browse files
committed
build: restore android savedmodel deps
1 parent 9d72f2f commit 5ec1920

1 file changed

Lines changed: 38 additions & 1 deletion

File tree

scripts/patch_tfjava.py

Lines changed: 38 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1313,7 +1313,44 @@ def patch_workspace(path: Path) -> None:
13131313

13141314
# Keep upstream SavedModel loader wiring on Android so the C API can resolve
13151315
# TF_LoadSessionFromSavedModel() once the mobile guard above is removed.
1316-
SAVED_MODEL_ANDROID_LOADER_PATCH = ""
1316+
SAVED_MODEL_ANDROID_LOADER_PATCH = """--- a/tensorflow/cc/saved_model/BUILD
1317+
+++ b/tensorflow/cc/saved_model/BUILD
1318+
@@ -50,6 +50,9 @@
1319+
# tf_lib depending on the build platform.
1320+
"//tensorflow/core:lib",
1321+
"//tensorflow/core:protos_all_cc",
1322+
+ ]) + if_android([
1323+
+ "//tensorflow/core:lib",
1324+
+ "//tensorflow/core:protos_all_cc",
1325+
]),
1326+
)
1327+
1328+
@@ -98,6 +101,10 @@
1329+
"//tensorflow/core:core_cpu",
1330+
"//tensorflow/core:lib",
1331+
"//tensorflow/core:protos_all_cc",
1332+
+ ]) + if_android([
1333+
+ "//tensorflow/core:core_cpu",
1334+
+ "//tensorflow/core:lib",
1335+
+ "//tensorflow/core:protos_all_cc",
1336+
]),
1337+
)
1338+
1339+
@@ -109,6 +116,13 @@
1340+
":constants",
1341+
":reader",
1342+
] + if_not_mobile([
1343+
+ "//tensorflow/core:core_cpu",
1344+
+ "//tensorflow/core:framework",
1345+
+ "//tensorflow/core:lib",
1346+
+ "//tensorflow/core:lib_internal",
1347+
+ "//tensorflow/core:protos_all_cc",
1348+
+ "//tensorflow/core/util/tensor_bundle:naming",
1349+
+ ]) + if_android([
1350+
"//tensorflow/core:core_cpu",
1351+
"//tensorflow/core:framework",
1352+
"//tensorflow/core:lib",
1353+
"""
13171354

13181355

13191356
# Keep upstream tensorflow_framework deps on Android so loader_lite_impl stays

0 commit comments

Comments
 (0)