diff options
author | Sourabh Bajaj <sourabhbajaj@google.com> | 2018-01-17 13:14:48 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-17 13:18:49 -0800 |
commit | d8697935d334bb0f2e1c9bccfe9a2a7bee9785cc (patch) | |
tree | 134cec2030a5e3fd3edef20e739c133c8940f257 /tensorflow/java/BUILD | |
parent | 7699ea8bee2ee9d0d5c93706cf1357a36c60ef58 (diff) |
Merge changes from github.
PiperOrigin-RevId: 182258809
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD index bd60f151de..5a533e3b60 100644 --- a/tensorflow/java/BUILD +++ b/tensorflow/java/BUILD @@ -14,6 +14,7 @@ load( "tf_copts", "tf_custom_op_library", "tf_java_test", + "tf_cc_test", ) java_library( @@ -113,10 +114,12 @@ cc_library( name = "java_op_gen_lib", srcs = [ "src/gen/cc/op_generator.cc", + "src/gen/cc/source_writer.cc", ], hdrs = [ "src/gen/cc/java_defs.h", "src/gen/cc/op_generator.h", + "src/gen/cc/source_writer.h", ], copts = tf_copts(), deps = [ @@ -305,6 +308,20 @@ filegroup( ]), ) +tf_cc_test( + name = "source_writer_test", + size = "small", + srcs = [ + "src/gen/cc/source_writer_test.cc", + ], + deps = [ + ":java_op_gen_lib", + "//tensorflow/core:lib", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + ], +) + filegroup( name = "libtensorflow_jni", srcs = select({ |