diff options
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 12 |
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"], |