aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/java/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r--tensorflow/java/BUILD12
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD
index f2904ad5a6..a8910248c1 100644
--- a/tensorflow/java/BUILD
+++ b/tensorflow/java/BUILD
@@ -55,6 +55,18 @@ java_test(
)
java_test(
+ name = "OperationTest",
+ size = "small",
+ srcs = ["src/test/java/org/tensorflow/OperationTest.java"],
+ test_class = "org.tensorflow.OperationTest",
+ deps = [
+ ":tensorflow",
+ ":testutil",
+ "@junit",
+ ],
+)
+
+java_test(
name = "SavedModelBundleTest",
size = "small",
srcs = ["src/test/java/org/tensorflow/SavedModelBundleTest.java"],