experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

FIx CudnnCompatibleLSTM document error

Browse Source 
Fix the CudnnCompatibleLSTM document error, it should be tf.contrib.cudnn_rnn.CudnnCompatibleLSTMCell instead of tf.contrib.cudnn_rnn.CudnnCompatibleLSTM.
This commit is contained in:
Jiankang 2019-02-22 15:05:55 +08:00 committed by GitHub
parent 11f4ecb547
commit 27810c92b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py
View File

@ -104,7 +104,7 @@ class _CudnnRNN(base_layer.Layer):
# Inference subgraph for unidirectional RNN on, e.g., CPU or mobile. # Inference subgraph for unidirectional RNN on, e.g., CPU or mobile.
with tf.Graph().as_default(): with tf.Graph().as_default():
single_cell = lambda: tf.contrib.cudnn_rnn.CudnnCompatibleLSTM(num_units) single_cell = lambda: tf.contrib.cudnn_rnn.CudnnCompatibleLSTMCell(num_units)
# NOTE: Even if there's only one layer, the cell needs to be wrapped in # NOTE: Even if there's only one layer, the cell needs to be wrapped in
# MultiRNNCell. # MultiRNNCell.
@ -124,7 +124,7 @@ class _CudnnRNN(base_layer.Layer):
# Inference subgraph for bidirectional RNN # Inference subgraph for bidirectional RNN
with tf.Graph().as_default(): with tf.Graph().as_default():
single_cell = lambda: tf.contrib.cudnn_rnn.CudnnCompatibleLSTM(num_units) single_cell = lambda: tf.contrib.cudnn_rnn.CudnnCompatibleLSTMCell(num_units)
cells_fw = [single_cell() for _ in range(num_layers)] cells_fw = [single_cell() for _ in range(num_layers)]
cells_bw = [single_cell() for _ in range(num_layers)] cells_bw = [single_cell() for _ in range(num_layers)]