diff options
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({ |