diff options
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD index 9de79af7d2..a380bc2c71 100644 --- a/tensorflow/java/BUILD +++ b/tensorflow/java/BUILD @@ -10,8 +10,9 @@ load(":src/gen/gen_ops.bzl", "tf_java_op_gen_srcjar") load( "//tensorflow:tensorflow.bzl", "tf_binary_additional_srcs", - "tf_copts", "tf_cc_binary", + "tf_copts", + "tf_custom_op_library", "tf_java_test", ) @@ -180,10 +181,16 @@ tf_java_test( ], ) +tf_custom_op_library( + name = "my_test_op.so", + srcs = ["src/test/native/my_test_op.cc"], +) + tf_java_test( name = "TensorFlowTest", size = "small", srcs = ["src/test/java/org/tensorflow/TensorFlowTest.java"], + data = [":my_test_op.so"], javacopts = JAVACOPTS, test_class = "org.tensorflow.TensorFlowTest", deps = [ |