W3cubDocs

/TensorFlow Python

Module: tf.contrib.cudnn_rnn

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 CudnnRNNRelu: Cudnn implementation of the RNN-relu layer.

class CudnnRNNReluSaveable: SaveableObject implementation handling Cudnn RNN Relu opaque params.

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

class CudnnRNNTanhSaveable: SaveableObject implementation handling Cudnn RNN Tanh opaque params.

Other Members

__cached__

__loader__

__spec__

© 2018 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/api_docs/python/tf/contrib/cudnn_rnn