aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/seq2seq/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/seq2seq/__init__.py')
-rw-r--r--tensorflow/contrib/seq2seq/__init__.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/contrib/seq2seq/__init__.py b/tensorflow/contrib/seq2seq/__init__.py
index e67e4a7ca9..2627e31426 100644
--- a/tensorflow/contrib/seq2seq/__init__.py
+++ b/tensorflow/contrib/seq2seq/__init__.py
@@ -13,7 +13,7 @@
# limitations under the License.
# ==============================================================================
-"""Ops for building neural network seq2seq layers and losses."""
+"""Ops for building neural network seq2seq decoders and losses."""
from __future__ import absolute_import
from __future__ import division
@@ -22,6 +22,7 @@ from __future__ import print_function
import sys
# pylint: disable=unused-import,line-too-long
-from tensorflow.contrib.seq2seq.python.ops import layers
-from tensorflow.contrib.seq2seq.python.ops import loss
+from tensorflow.contrib.seq2seq.python.ops.decoder_fn import *
+from tensorflow.contrib.seq2seq.python.ops.loss import *
+from tensorflow.contrib.seq2seq.python.ops.seq2seq import *
# pylint: enable=unused-import,line-too-long