diff options
| author |  Yao Zhang <yaozhang@google.com> | 2017-03-23 12:02:50 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-23 13:31:24 -0700 |
| commit | 8ca071456537e6c96ae8896c2a20b1f08b0e59d3 (patch) | |
| tree | 245822721c30e46cc495314fa6de64357564c5e9 | |
| parent | 26a244c8f8c0352cc8f90e4e4af2dcc314279b93 (diff) | |
Target devices should use tf_cuda_library.
Change: 151042735
| -rw-r--r-- | tensorflow/core/grappler/BUILD | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/BUILD b/tensorflow/core/grappler/BUILD index c3065df27b..66cbbf1c1f 100644 --- a/tensorflow/core/grappler/BUILD +++ b/tensorflow/core/grappler/BUILD @@ -2,6 +2,8 @@ licenses(["notice"]) # Apache 2.0 exports_files(["LICENSE"]) +load("//tensorflow:tensorflow.bzl", "tf_cuda_library") + filegroup( name = "all_files", srcs = glob( @@ -51,7 +53,7 @@ cc_test( ], ) -cc_library( +tf_cuda_library( name = "devices", srcs = ["devices.cc"], hdrs = ["devices.h"], |