We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ca181a5 commit f6598a1Copy full SHA for f6598a1
1 file changed
scripts/patch_tfjava.py
@@ -920,6 +920,20 @@ def patch_workspace(path: Path) -> None:
920
"//conditions:default": [
921
":tf_attrtype",
922
"//tensorflow/core:core_cpu",
923
+@@ -149,10 +154,15 @@ tf_cuda_library(
924
+ srcs = [
925
+ "c_api.cc",
926
+ "c_api_function.cc",
927
+- ],
928
++ ] + select({
929
++ "//tensorflow:android": ["//tensorflow/cc/saved_model:loader.h"],
930
++ "//conditions:default": [],
931
++ }),
932
+ hdrs = [
933
+ "c_api.h",
934
+ ],
935
+ copts = tf_copts(),
936
+ visibility = ["//tensorflow/c:__subpackages__"],
937
@@ -162,7 +167,23 @@ tf_cuda_library(
938
":tf_status_internal",
939
] + select({
0 commit comments