From c5110186a2df59a5da376914b70dd0f124dfddc6 Mon Sep 17 00:00:00 2001 From: gangliao Date: Mon, 23 Oct 2017 15:48:37 +0800 Subject: [PATCH] Fix import mpi_collectives as mpi --- tensorflow/contrib/mpi_collectives/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/contrib/mpi_collectives/__init__.py b/tensorflow/contrib/mpi_collectives/__init__.py index 20d6ddaf8..8f883d0b7 100644 --- a/tensorflow/contrib/mpi_collectives/__init__.py +++ b/tensorflow/contrib/mpi_collectives/__init__.py @@ -37,7 +37,7 @@ Example: ```python -from tensorflow.contrib import mpi +from tensorflow.contrib import mpi_collectives as mpi # Use `mpi.Session` instead of `tf.Session` with mpi.Session() as session: