diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-11 23:48:06 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-11 23:50:49 -0700 |
| commit | a436cf493d3a590572aec9fe574f0e9028e8b61e (patch) | |
| tree | 2a72dc460458b90a27f1075afbb01feee118ac59 /tensorflow/contrib/fused_conv | |
| parent | fc5250f97188e9b247845e32692d1c4ffad170c4 (diff) | |
Adding cuDNN header dependency to targets that include the cuDNN header file.
PiperOrigin-RevId: 196349902
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", ], ) |