Module: tfa.rnn

View source on GitHub

Additional RNN cells that corform to Keras API.


esn_cell module: Implements ESN Cell.

layer_norm_lstm_cell module: Implements LayerNormLSTM Cell.

layer_norm_simple_rnn_cell module: Implements LayerNormSimpleRNNCell Cell.

nas_cell module: Implements NAS Cell.

peephole_lstm_cell module: Implements PeepholeLSTM Cell.


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.