diff options
author | Mingsheng Hong <hongm@google.com> | 2018-02-14 13:51:14 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-14 13:56:45 -0800 |
commit | 21fe8feb34cb4d5b15ce35fc421c7430307facd2 (patch) | |
tree | 6a5972f199c40521ddbc2f3ade92df8c4082d574 /tensorflow/c/BUILD | |
parent | 620dc3f097d047346943c416823f5e370df9fe4b (diff) |
Added C-API based unit tests for GPU and XLA GPU testing.
Also refined the API comment for TF_NewSession().
PiperOrigin-RevId: 185739196
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 25a994be3e..9060c58c13 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -6,6 +6,7 @@ licenses(["notice"]) # Apache 2.0 load( "//tensorflow:tensorflow.bzl", "tf_cc_test", + "tf_cuda_cc_test", "tf_copts", "tf_cuda_library", "tf_custom_op_library", @@ -155,7 +156,7 @@ tf_cuda_library( ], ) -tf_cc_test( +tf_cuda_cc_test( name = "c_api_test", size = "small", srcs = ["c_api_test.cc"], |