Skip to content

Commit f6598a1

Browse files
committed
build: declare savedmodel loader header
1 parent ca181a5 commit f6598a1

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

scripts/patch_tfjava.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -920,6 +920,20 @@ def patch_workspace(path: Path) -> None:
920920
"//conditions:default": [
921921
":tf_attrtype",
922922
"//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__"],
923937
@@ -162,7 +167,23 @@ tf_cuda_library(
924938
":tf_status_internal",
925939
] + select({

0 commit comments

Comments
 (0)