Defined in tensorflow/contrib/cudnn_rnn/__init__.py
.
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
: SaveableObject implementation handling Cudnn LSTM opaque params.
class CudnnRNNTanh
: Cudnn implementation of the RNN-tanh layer.
class CudnnRNNTanhSaveable
: SaveableObject implementation handling Cudnn LSTM opaque params.