Модуль: tf.contrib.cudnn_rnn
Операции для объединённых моделей Cudnn RNN.
Классы
class CudnnCompatibleGRUCell: Совместимая с Cudnn ячейка GRU.
class CudnnCompatibleLSTMCell: Совместимая с Cudnn ячейка LSTM.
class CudnnGRU: Реализация слоя GRU с помощью Cudnn.
class CudnnGRUSaveable: Реализация SaveableObject, обрабатывающая непрозрачные параметры Cudnn GRU.
class CudnnLSTM: Реализация слоя LSTM с помощью Cudnn.
class CudnnLSTMSaveable: Реализация SaveableObject, обрабатывающая непрозрачные параметры Cudnn LSTM.
class CudnnParamsFormatConverterGRU: Вспомогательный класс, преобразующий параметры между Cudnn и TF GRU.
class CudnnParamsFormatConverterLSTM: Вспомогательный класс, преобразующий параметры между Cudnn и TF LSTM.
class CudnnParamsFormatConverterRelu: Вспомогательный класс, преобразующий параметры между Cudnn и TF Relu RNN.
class CudnnParamsFormatConverterTanh: Вспомогательный класс, преобразующий параметры между Cudnn и TF Tanh RNN.
class CudnnRNNRelu: Реализация слоя RNN-relu с помощью Cudnn.
class CudnnRNNReluSaveable: Реализация SaveableObject, обрабатывающая непрозрачные параметры Cudnn LSTM.
class CudnnRNNTanh: Реализация слоя RNN-tanh с помощью Cudnn.
class CudnnRNNTanhSaveable: Реализация SaveableObject, обрабатывающая непрозрачные параметры Cudnn LSTM.
© 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