diff options
author | Asim Shankar <ashankar@google.com> | 2018-05-29 00:50:34 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-29 00:53:00 -0700 |
commit | 8ecf1ebc5d83e66b29a07113b53c49ef8264703c (patch) | |
tree | 5b6c00f7ae5f1beeae52482bb5411057d4735fff /tensorflow/java/BUILD | |
parent | 23c70a240219da7201806e4ebc1354f568581557 (diff) |
Re-apply #18192.
PiperOrigin-RevId: 198358055
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 41 |
1 files changed, 14 insertions, 27 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD index 78596d147a..19d2133a55 100644 --- a/tensorflow/java/BUILD +++ b/tensorflow/java/BUILD @@ -60,9 +60,7 @@ java_library( filegroup( name = "java_op_sources", - srcs = glob(["src/main/java/org/tensorflow/op/**/*.java"]) + [ - ":java_op_gen_sources", - ], + srcs = glob(["src/main/java/org/tensorflow/op/**/*.java"]) + [":java_op_gen_sources"], visibility = [ "//tensorflow/java:__pkg__", ], @@ -70,43 +68,27 @@ filegroup( tf_java_op_gen_srcjar( name = "java_op_gen_sources", - gen_base_package = "org.tensorflow.op", - gen_tool = "java_op_gen_tool", - ops_libs = [ - "array_ops", - "candidate_sampling_ops", - "control_flow_ops", - "data_flow_ops", - "image_ops", - "io_ops", - "linalg_ops", - "logging_ops", - "math_ops", - "nn_ops", - "no_op", - "parsing_ops", - "random_ops", - "sparse_ops", - "state_ops", - "string_ops", - "training_ops", - "user_ops", + api_def_srcs = [ + "//tensorflow/core/api_def:base_api_def", ], + base_package = "org.tensorflow.op", + gen_tool = ":java_op_gen_tool", ) -# Build the gen tool as a library, as it will be linked to a core/ops binary -# file before making it an executable. See tf_java_op_gen_srcjar(). -cc_library( +tf_cc_binary( name = "java_op_gen_tool", srcs = [ "src/gen/cc/op_gen_main.cc", ], copts = tf_copts(), + linkopts = ["-lm"], + linkstatic = 1, deps = [ ":java_op_gen_lib", "//tensorflow/core:framework", "//tensorflow/core:framework_internal", "//tensorflow/core:lib", + "//tensorflow/core:ops", ], ) @@ -114,11 +96,13 @@ cc_library( name = "java_op_gen_lib", srcs = [ "src/gen/cc/op_generator.cc", + "src/gen/cc/op_specs.cc", "src/gen/cc/source_writer.cc", ], hdrs = [ "src/gen/cc/java_defs.h", "src/gen/cc/op_generator.h", + "src/gen/cc/op_specs.h", "src/gen/cc/source_writer.h", ], copts = tf_copts(), @@ -127,6 +111,9 @@ cc_library( "//tensorflow/core:framework_internal", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", + "//tensorflow/core:op_gen_lib", + "//tensorflow/core:protos_all_cc", + "@com_googlesource_code_re2//:re2", ], ) |