diff options
author | 2017-10-03 21:37:43 -0700 | |
---|---|---|
committer | 2017-10-03 21:43:50 -0700 | |
commit | f9f037c1c489d6a72ef682e3bce01e6f154222e4 (patch) | |
tree | 0bb6bbc28f0b08e82ddb8d3a9d299904c973413a /tensorflow/contrib/cudnn_rnn | |
parent | bfaaefa9ecbbbc797f5af60f3d87f6a3c3ac7a09 (diff) |
Bugfix to LSTMBlockCell and friends: clipping is off by default.
* Rename broken API argu clip_cell boolean to cell_clip value.
* Make default no clipping.
PiperOrigin-RevId: 170960975
Diffstat (limited to 'tensorflow/contrib/cudnn_rnn')
-rw-r--r-- | tensorflow/contrib/cudnn_rnn/python/ops/cudnn_rnn_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/cudnn_rnn/python/ops/cudnn_rnn_ops.py b/tensorflow/contrib/cudnn_rnn/python/ops/cudnn_rnn_ops.py index f6eeb01675..bbf1bd9bca 100644 --- a/tensorflow/contrib/cudnn_rnn/python/ops/cudnn_rnn_ops.py +++ b/tensorflow/contrib/cudnn_rnn/python/ops/cudnn_rnn_ops.py @@ -65,7 +65,7 @@ class CudnnCompatibleLSTMCell(lstm_ops.LSTMBlockCell): def __init__(self, num_units, reuse=None): super(CudnnCompatibleLSTMCell, self).__init__( - num_units, forget_bias=0, clip_cell=False, use_peephole=False, + num_units, forget_bias=0, cell_clip=None, use_peephole=False, reuse=reuse) self._names.update({"scope": "cudnn_compatible_lstm_cell"}) |