diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-26 23:58:04 -0700 |
|---|---|---|
| committer |  Gunhan Gulsoy <gunan@google.com> | 2018-06-28 21:37:43 -0700 |
| commit | 11157efc4e94a7c70ff7532d7bb835fb5d9d19da (patch) | |
| tree | 492384123eabb96505a68c65c77dcc28094d1c45 /tensorflow/contrib/BUILD | |
| parent | c30c57bd0792c50397883252ee5b2960988846d3 (diff) | |
Fix Windows GPU Build
PiperOrigin-RevId: 202260254
Diffstat (limited to 'tensorflow/contrib/BUILD')
| -rw-r--r-- | tensorflow/contrib/BUILD | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index fffab5a795..2d7916c8b1 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -9,6 +9,7 @@ load("//third_party/mpi:mpi.bzl", "if_mpi") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt") load("//tensorflow:tensorflow.bzl", "if_not_windows") +load("//tensorflow:tensorflow.bzl", "if_not_windows_cuda") py_library( name = "contrib_py", @@ -45,7 +46,6 @@ py_library( "//tensorflow/contrib/factorization:factorization_py", "//tensorflow/contrib/feature_column:feature_column_py", "//tensorflow/contrib/framework:framework_py", - "//tensorflow/contrib/fused_conv:fused_conv_py", "//tensorflow/contrib/gan", "//tensorflow/contrib/graph_editor:graph_editor_py", "//tensorflow/contrib/grid_rnn:grid_rnn_py", @@ -123,7 +123,9 @@ py_library( "//tensorflow/contrib/kafka", ], "//conditions:default": [], - }) + if_not_windows([ + }) + if_not_windows_cuda([ + "//tensorflow/contrib/fused_conv:fused_conv_py", # unresolved symbols, need to export more symbols + ]) + if_not_windows([ "//tensorflow/contrib/ffmpeg:ffmpeg_ops_py", "//tensorflow/contrib/lite/python:lite", # unix dependency, need to fix code ]), |