diff options
Diffstat (limited to 'tensorflow/contrib/BUILD')
| -rw-r--r-- | tensorflow/contrib/BUILD | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 798f499870..d98a24994c 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -166,7 +166,9 @@ cc_library( "//tensorflow/contrib/kinesis:dataset_kernels", ], "//conditions:default": [], - }), + }) + if_not_windows([ + "//tensorflow/contrib/tensorrt:trt_engine_op_kernel", + ]), ) cc_library( @@ -203,5 +205,7 @@ cc_library( "//tensorflow/contrib/kinesis:dataset_ops_op_lib", ], "//conditions:default": [], - }), + }) + if_not_windows([ + "//tensorflow/contrib/tensorrt:trt_engine_op_op_lib", + ]), ) |