Module: tfa.rnn

Additional RNN cells that corform to Keras API.


class ESNCell: Echo State recurrent Network (ESN) cell.

class LayerNormLSTMCell: LSTM cell with layer normalization and recurrent dropout.

class LayerNormSimpleRNNCell: Cell class for LayerNormSimpleRNN.

class NASCell: Neural Architecture Search (NAS) recurrent network cell.

class PeepholeLSTMCell: Equivalent to tf.keras.layers.LSTMCell class but adds peephole connections.