diff options
| author |  Scott Zhu <scottzhu@google.com> | 2018-04-13 17:52:20 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-13 17:57:27 -0700 |
| commit | 3652556dab3ebfe0152232facc7304fe5754aecb (patch) | |
| tree | 9a9cecde4c85dc53548a185f9bd6d7c6e0591262 /tensorflow/contrib/rnn | |
| parent | ef24ad14502e992716c49fdd5c63e6b2c2fb6b5a (diff) | |
Merge changes from github.
PiperOrigin-RevId: 192850372
Diffstat (limited to 'tensorflow/contrib/rnn')
| -rw-r--r-- | tensorflow/contrib/rnn/python/ops/rnn_cell.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/rnn/python/ops/rnn_cell.py b/tensorflow/contrib/rnn/python/ops/rnn_cell.py index 2f6ae9f367..b12e2cd5ed 100644 --- a/tensorflow/contrib/rnn/python/ops/rnn_cell.py +++ b/tensorflow/contrib/rnn/python/ops/rnn_cell.py @@ -2891,7 +2891,7 @@ class WeightNormLSTMCell(rnn_cell_impl.RNNCell): output_size = weight.get_shape().as_list()[1] g = vs.get_variable(name, [output_size], dtype=weight.dtype) - return nn_impl.l2_normalize(weight, dim=0) * g + return nn_impl.l2_normalize(weight, axis=0) * g def _linear(self, args, |