Module: tf.contrib.cudnn_rnn

Module tf.contrib.cudnn_rnn

Defined in tensorflow/contrib/cudnn_rnn/__init__.py.

Ops for fused Cudnn RNN models.

Classes

class CudnnGRU: Cudnn implementation of the GRU model.

class CudnnLSTM: Cudnn implementation of the LSTM model.

class CudnnRNNRelu: Cudnn implementation of the RNN-relu model.

class CudnnRNNTanh: Cudnn implementation of the RNN-tanh model.

class RNNParamsSaveable: SaveableObject implementation that handles the RNN params variable.