diff options
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD index 78ef152689..f2904ad5a6 100644 --- a/tensorflow/java/BUILD +++ b/tensorflow/java/BUILD @@ -55,6 +55,19 @@ java_test( ) java_test( + name = "SavedModelBundleTest", + size = "small", + srcs = ["src/test/java/org/tensorflow/SavedModelBundleTest.java"], + data = ["//tensorflow/cc/saved_model:saved_model_half_plus_two"], + test_class = "org.tensorflow.SavedModelBundleTest", + deps = [ + ":tensorflow", + ":testutil", + "@junit", + ], +) + +java_test( name = "SessionTest", size = "small", srcs = ["src/test/java/org/tensorflow/SessionTest.java"], |