Module: tf.contrib.cudnn_rnn

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/cudnn_rnn