Module: tfa.rnn

Additional RNN cells that corform to Keras API.

Classes

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.