aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/java/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r--tensorflow/java/BUILD26
1 files changed, 25 insertions, 1 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD
index 9abb63c966..90372660cd 100644
--- a/tensorflow/java/BUILD
+++ b/tensorflow/java/BUILD
@@ -9,7 +9,10 @@ load("build_defs", "JAVACOPTS")
java_library(
name = "tensorflow",
- srcs = [":java_sources"],
+ srcs = [
+ ":java_op_sources",
+ ":java_sources",
+ ],
data = [":libtensorflow_jni"],
javacopts = JAVACOPTS,
visibility = ["//visibility:public"],
@@ -26,6 +29,14 @@ filegroup(
],
)
+filegroup(
+ name = "java_op_sources",
+ srcs = glob(["src/main/java/org/tensorflow/op/*.java"]),
+ visibility = [
+ "//tensorflow/java:__pkg__",
+ ],
+)
+
java_library(
name = "testutil",
testonly = 1,
@@ -138,6 +149,19 @@ java_test(
],
)
+java_test(
+ name = "ScopeTest",
+ size = "small",
+ srcs = ["src/test/java/org/tensorflow/op/ScopeTest.java"],
+ javacopts = JAVACOPTS,
+ test_class = "org.tensorflow.op.ScopeTest",
+ deps = [
+ ":tensorflow",
+ ":testutil",
+ "@junit",
+ ],
+)
+
filegroup(
name = "libtensorflow_jni",
srcs = select({