diff --git a/tensorflow/contrib/rnn/ops/lstm_ops.cc b/tensorflow/contrib/rnn/ops/lstm_ops.cc index 2de40825c90..699cc6c88a4 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 ```