tf.raw_ops.LSTMBlockCellGrad

Computes the LSTM cell backward propagation for 1 timestep.

This implementation is to be used in conjunction of LSTMBlockCell.

Args
x A Tensor. Must be one of the following types: half, float32. The input to the LSTM cell, shape (batch_size, num_inputs).
cs_prev A Tensor. Must have the same type as x. The previous cell state.
h_prev A Tensor. Must have the same type as x. The previous h state.
w A Tensor. Must have the same type as x. The weight matrix.
wci A Tensor. Must have the same type as x. The weight matrix for input gate peephole connection.
wcf A Tensor. Must have the same type as x. The weight matrix for forget gate peephole connection.
wco A Tensor. Must have the same type as x. The weight matrix for output gate peephole connection.
b A Tensor. Must have the same type as x. The bias vector.
i A Tensor. Must have the same type as x. The input gate.
cs A Tensor. Must have the same type as x. The cell state before the tanh.
f A Tensor. Must have the same type as x. The forget gate.
o A Tensor. Must have the same type as x. The output gate.
ci A Tensor. Must have the same type as x. The cell input.
co A Tensor. Must have the same type as x. The cell after the tanh.
cs_grad A Tensor. Must have the same type as x. The current gradient of cs.
h_grad A Tensor. Must have the same type as x. The gradient of h vector.
use_peephole A bool. Whether the cell uses peephole connections.
name A name for the operation (optional).
Returns
A tuple of Tensor objects (cs_prev_grad, dicfo, wci_grad, wcf_grad, wco_grad).
cs_prev_grad A Tensor. Has the same type as x.
dicfo A Tensor. Has the same type as x.
wci_grad A Tensor. Has the same type as x.
wcf_grad A Tensor. Has the same type as x.
wco_grad A Tensor. Has the same type as x.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/LSTMBlockCellGrad