diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2017-07-17 08:29:52 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-17 08:33:52 -0700 |
commit | bed7cbe389420abe760d86cd345e88d45eb624ab (patch) | |
tree | c4d5c9d789b06e7596b6473902c140638fbb5de7 /tensorflow/c/BUILD | |
parent | a925c14596135b19bedb73e0f6ae3cd170180106 (diff) |
Split out new while_loop_test.cc from c_api_test.cc
This change also separates shared functionality into
c_test_util.h/cc. This brings c_api_test.cc to a mere 1715 LOC
(further splits can be more easily done now too).
PiperOrigin-RevId: 162216399
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index af9071238b..507b2fe1f1 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -103,6 +103,19 @@ tf_cuda_library( # ----------------------------------------------------------------------------- # Tests +tf_cuda_library( + name = "c_test_util", + testonly = 1, + srcs = ["c_test_util.cc"], + hdrs = ["c_test_util.h"], + deps = [ + ":c_api", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + ], +) + tf_cc_test( name = "c_api_test", size = "small", @@ -120,6 +133,7 @@ tf_cc_test( # linkstatic = tf_kernel_tests_linkstatic(), deps = [ ":c_api", + ":c_test_util", "//tensorflow/cc:cc_ops", "//tensorflow/cc:grad_ops", "//tensorflow/cc/saved_model:signature_constants", @@ -139,6 +153,19 @@ tf_cc_test( ], ) +tf_cc_test( + name = "while_loop_test", + size = "small", + srcs = ["while_loop_test.cc"], + deps = [ + ":c_api", + ":c_test_util", + "//tensorflow/core:lib", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + ], +) + tf_custom_op_library( name = "test_op.so", srcs = ["test_op.cc"], |