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: 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.