aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/__init__.py
diff options
context:
space:
mode:
authorGravatar Francois Chollet <fchollet@google.com>2018-04-10 16:11:38 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-10 16:14:13 -0700
commitdc8aa019ba27d65789bcecbc776d1ccc9359c011 (patch)
tree0063bfbc217a96278f90f14ccf6d5f08ec33cc9a /tensorflow/python/__init__.py
parent47d72205f3c58d31bfec52eb331e89edc562106c (diff)
Fix `nn` module RNN namespace issues.
PiperOrigin-RevId: 192364808
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r--tensorflow/python/__init__.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py
index da836aca6f..13f8420a67 100644
--- a/tensorflow/python/__init__.py
+++ b/tensorflow/python/__init__.py
@@ -157,6 +157,9 @@ from tensorflow.python.ops import rnn_cell
# Required due to `rnn` and `rnn_cell` not being imported in `nn` directly
# (due to a circular dependency issue: rnn depends on layers).
nn.dynamic_rnn = rnn.dynamic_rnn
+nn.static_rnn = rnn.static_rnn
+nn.raw_rnn = rnn.raw_rnn
+nn.bidirectional_dynamic_rnn = rnn.bidirectional_dynamic_rnn
nn.rnn_cell = rnn_cell
# Symbols whitelisted for export without documentation.