diff options
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index f745c175b1..738b74c929 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -105,7 +105,7 @@ py_library( "//tensorflow/contrib/training:training_py", "//tensorflow/contrib/util:util_py", "//tensorflow/python:util", - ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_ops_py"]) + ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + if_tensorrt(["//tensorflow/contrib/tensorrt:init_py"]), ) |