diff options
Diffstat (limited to 'tensorflow/contrib/rnn/ops/lstm_ops.cc')
| -rw-r--r-- | tensorflow/contrib/rnn/ops/lstm_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/rnn/ops/lstm_ops.cc b/tensorflow/contrib/rnn/ops/lstm_ops.cc index 2de40825c9..699cc6c88a 100644 --- a/tensorflow/contrib/rnn/ops/lstm_ops.cc +++ b/tensorflow/contrib/rnn/ops/lstm_ops.cc @@ -78,7 +78,7 @@ ci = tanh(ci) cs = ci .* i + cs_prev .* f cs = clip(cs, cell_clip) -o = sigmoid(cs * wco + f) +o = sigmoid(cs * wco + o) co = tanh(cs) h = co .* o ``` |