add tf-c-header rule (tensorflow#1244)

i-ony · Mar 15, 2021 · b5bb3ea · b5bb3ea
1 parent 86cb167
commit b5bb3ea
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/third_party/toolchains/tf/BUILD.tpl b/third_party/toolchains/tf/BUILD.tpl
@@ -23,6 +23,14 @@ cc_library(
     visibility = ["//visibility:public"],
 )
 
+cc_library(
+    name = "tf_c_header_lib",
+    hdrs = [":tf_c_header_include"],
+    include_prefix = "tensorflow/c",
+    strip_include_prefix = "include_c",
+    visibility = ["//visibility:public"],
+)
+
 cc_library(
     name = "libtensorflow_framework",
     srcs = [":libtensorflow_framework.so"],