diff options
author | 2016-08-03 05:49:13 -0800 | |
---|---|---|
committer | 2016-08-03 07:03:38 -0700 | |
commit | 9f400c6ec37ec1d411f106e5821b69a1a46c16b0 (patch) | |
tree | 167855392dcac76c0bf13230564642802a89dce3 | |
parent | e4e49b056e0128e057c4440732ccffa01c5c13f5 (diff) |
Give tf.contrib.rnn fused RNN cell public visibility.
Change: 129209225
-rw-r--r-- | tensorflow/contrib/rnn/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/contrib/rnn/python/ops/lstm_ops.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/rnn/BUILD b/tensorflow/contrib/rnn/BUILD index dffd139ec0..f69c656c68 100644 --- a/tensorflow/contrib/rnn/BUILD +++ b/tensorflow/contrib/rnn/BUILD @@ -18,6 +18,7 @@ py_library( ":python/ops/_lstm_ops.so", ], srcs_version = "PY2AND3", + visibility = ["//visibility:public"], ) cuda_py_tests( diff --git a/tensorflow/contrib/rnn/python/ops/lstm_ops.py b/tensorflow/contrib/rnn/python/ops/lstm_ops.py index 2ecc415d35..231d13caa6 100644 --- a/tensorflow/contrib/rnn/python/ops/lstm_ops.py +++ b/tensorflow/contrib/rnn/python/ops/lstm_ops.py @@ -411,7 +411,7 @@ class LSTMFusedCell(rnn_cell.RNNCell): Args: num_units: int, The number of units in the LSTM cell. forget_bias: float, The bias added to forget gates (see above). - use_peephole: Whether to use peephole connectios or not. + use_peephole: Whether to use peephole connections or not. """ self._num_units = num_units self._forget_bias = forget_bias |