diff options
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 7d44a054a8..fffab5a795 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -114,6 +114,7 @@ py_library( "//tensorflow/contrib/training:training_py", "//tensorflow/contrib/util:util_py", "//tensorflow/python:util", + "//tensorflow/python/estimator:estimator_py", ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + if_tensorrt([ "//tensorflow/contrib/tensorrt:init_py", ]) + select({ |