TensorFlow 2.0 Beta is available Learn more

Module: tf.contrib.cudnn_rnn

View source on GitHub

Ops for fused Cudnn RNN models.

Classes

class CudnnCompatibleGRUCell: Cudnn Compatible GRUCell.

class CudnnCompatibleLSTMCell: Cudnn Compatible LSTMCell.

class CudnnGRU: Cudnn implementation of the GRU layer.

class CudnnGRUSaveable: SaveableObject implementation handling Cudnn GRU opaque params.

class CudnnLSTM: Cudnn implementation of LSTM layer.

class CudnnLSTMSaveable: SaveableObject implementation handling Cudnn LSTM opaque params.

class CudnnParamsFormatConverterGRU: Helper class that converts between params of Cudnn and TF GRU.

class CudnnParamsFormatConverterLSTM: Helper class that converts between params of Cudnn and TF LSTM.

class CudnnParamsFormatConverterRelu: Helper class that converts between params of Cudnn and TF Relu RNN.

class CudnnParamsFormatConverterTanh: Helper class that converts between params of Cudnn and TF Tanh RNN.

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

class CudnnRNNReluSaveable

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

class CudnnRNNTanhSaveable