diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-01 07:30:28 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 07:32:52 -0700 |
| commit | 4349f663375ecbb7e678d1e86606380e42d431ae (patch) | |
| tree | b18b412942fafcd6f3eb9e385683e6b6a9049c5e /tensorflow/contrib/fused_conv | |
| parent | e6aca210f1082e4cb8cf3d0f775a79042b48f68a (diff) | |
Resubmitting CL 196349902: Adding cuDNN header dependency to targets that include the cuDNN header file.
PiperOrigin-RevId: 198869605
Diffstat (limited to 'tensorflow/contrib/fused_conv')
| -rw-r--r-- | tensorflow/contrib/fused_conv/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/fused_conv/BUILD b/tensorflow/contrib/fused_conv/BUILD index 0eb6889db1..0f0813c07f 100644 --- a/tensorflow/contrib/fused_conv/BUILD +++ b/tensorflow/contrib/fused_conv/BUILD @@ -75,6 +75,7 @@ tf_kernel_library( "//tensorflow/core/kernels:gpu_util_hdrs", "//tensorflow/core/kernels:ops_util_hdrs", "//third_party/eigen3", + "@local_config_cuda//cuda:cudnn_header", ], alwayslink = 1, ) @@ -94,6 +95,7 @@ tf_custom_op_library( "//tensorflow/core/kernels:conv_ops_gpu_hdrs", "//tensorflow/core/kernels:gpu_util_hdrs", "//tensorflow/core/kernels:ops_util_hdrs", + "@local_config_cuda//cuda:cudnn_header", ], ) |