aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r--tensorflow/c/BUILD7
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD
index 50887f70b5..7f86d345d7 100644
--- a/tensorflow/c/BUILD
+++ b/tensorflow/c/BUILD
@@ -7,6 +7,7 @@ load(
"//tensorflow:tensorflow.bzl",
"tf_cc_test",
"tf_cuda_library",
+ "tf_custom_op_library",
)
# For platform specific build config
@@ -60,6 +61,7 @@ tf_cc_test(
name = "c_api_test",
size = "small",
srcs = ["c_api_test.cc"],
+ data = [":test_op.so"],
linkopts = select({
"//tensorflow:darwin": ["-headerpad_max_install_names"],
"//conditions:default": [],
@@ -81,6 +83,11 @@ tf_cc_test(
],
)
+tf_custom_op_library(
+ name = "test_op.so",
+ srcs = ["test_op.cc"],
+)
+
# -----------------------------------------------------------------------------
# Google-internal targets.