tf.raw_ops.CudnnRNNV3

A RNN backed by cuDNN.

Computes the RNN from the input and initial states, with respect to the params buffer. Accepts one extra input "sequence_lengths" than CudnnRNN.

rnn_mode: Indicates the type of the RNN model. input_mode: Indicates whether there is a linear projection between the input and the actual computation before the first layer. 'skip_input' is only allowed when input_size == num_units; 'auto_select' implies 'skip_input' when input_size == num_units; otherwise, it implies 'linear_input'. direction: Indicates whether a bidirectional model will be used. Should be "unidirectional" or "bidirectional". dropout: Dropout probability. When set to 0., dropout is disabled. seed: The 1st part of a seed to initialize dropout. seed2: The 2nd part of a seed to initialize dropout. input: If time_major is true, this is a 3-D tensor with the shape of [seq_length, batch_size, input_size]. If time_major is false, the shape is [batch_size, seq_length, input_size]. input_h: If time_major is true, this is a 3-D tensor with the shape of [num_layer * dir, batch_size, num_units]. If time_major is false, the shape is [batch_size, num_layer * dir, num_units]. input_c: For LSTM, a 3-D tensor with the shape of [num_layer * dir, batch, num_units]. For other models, it is ignored. params: A 1-D tensor that contains the weights and biases in an opaque layout. The size must be created through CudnnRNNParamsSize, and initialized separately. Note that they might not be compatible across different generations. So it is a good idea to save and restore sequence_lengths: a vector of lengths of each input sequence. output: If time_major is true, this is a 3-D tensor with the shape of [seq_length, batch_size, dir * num_units]. If time_major is false, the shape is [batch_size, seq_length, dir * num_units]. output_h: The same shape has input_h. output_c: The same shape as input_c for LSTM. An empty tensor for other models. is_training: Indicates whether this operation is used for inference or training. time_major: Indicates whether the input/output format is time major or batch major. reserve_space: An opaque tensor that can be used in backprop calculation. It is only produced if is_training is true.

inputA Tensor. Must be one of the following types: bfloat16, half, float32, float64.
input_hA Tensor. Must have the same type as input.
input_cA Tensor. Must have the same type as input.
paramsA Tensor. Must have the same type as input.
sequence_lengthsA Tensor of type int32.
rnn_modeAn optional string from: "rnn_relu", "rnn_tanh", "lstm", "gru". Defaults to "lstm".
input_modeAn optional string from: "linear_input", "skip_input", "auto_select". Defaults to "linear_input".
directionAn optional string from: "unidirectional", "bidirectional". Defaults to "unidirectional".
dropoutAn optional float. Defaults to 0.
seedAn optional int. Defaults to 0.
seed2An optional int. Defaults to 0.
num_projAn optional int. Defaults to 0.
is_trainingAn optional bool. Defaults to True.
time_majorAn optional bool. Defaults to True.
nameA name for the operation (optional).

A tuple of Tensor objects (output, output_h, output_c, reserve_space, host_reserved).
outputA Tensor. Has the same type as input.
output_hA Tensor. Has the same type as input.
output_cA Tensor. Has the same type as input.
reserve_spaceA Tensor. Has the same type as input.
host_reservedA Tensor of type int8.