aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/java/BUILD
diff options
context:
space:
mode:
authorGravatar karl@kubx.ca <karl@kubx.ca>2018-05-13 23:46:37 -0400
committerGravatar karl@kubx.ca <karl@kubx.ca>2018-06-12 23:21:38 -0400
commit60552388401b3e70a21d4c01d3d374c9d85aea2b (patch)
treeabd9805d5f5fbac4e67f1764013a43ba8733a8d2 /tensorflow/java/BUILD
parent1aea422ca658d0ac2121245d31c3aa78a73c0efb (diff)
Complete operator processor for generating Ops API classes
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r--tensorflow/java/BUILD5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD
index 19d2133a55..47855c2d9b 100644
--- a/tensorflow/java/BUILD
+++ b/tensorflow/java/BUILD
@@ -56,6 +56,10 @@ java_library(
srcs = glob(["src/gen/java/org/tensorflow/processor/**/*.java"]),
javacopts = JAVACOPTS,
resources = glob(["src/gen/resources/META-INF/services/javax.annotation.processing.Processor"]),
+ deps = [
+ "@com_squareup_javapoet",
+ "@com_google_guava",
+ ],
)
filegroup(
@@ -70,6 +74,7 @@ tf_java_op_gen_srcjar(
name = "java_op_gen_sources",
api_def_srcs = [
"//tensorflow/core/api_def:base_api_def",
+ "//tensorflow/core/api_def:java_api_def",
],
base_package = "org.tensorflow.op",
gen_tool = ":java_op_gen_tool",