Unverified Commit 84840387 authored by Yuki Ueda's avatar Yuki Ueda Committed by GitHub
Browse files

Replace deprecated functions nn 2 rnn (#7338)

parent 0fb0fa84
...@@ -22,9 +22,9 @@ import tensorflow as tf ...@@ -22,9 +22,9 @@ import tensorflow as tf
# Supported rnn cells. # Supported rnn cells.
SUPPORTED_RNNS = { SUPPORTED_RNNS = {
"lstm": tf.nn.rnn_cell.BasicLSTMCell, "lstm": tf.contrib.rnn.BasicLSTMCell,
"rnn": tf.nn.rnn_cell.RNNCell, "rnn": tf.contrib.rnn.RNNCell,
"gru": tf.nn.rnn_cell.GRUCell, "gru": tf.contrib.rnn.GRUCell,
} }
# Parameters for batch normalization. # Parameters for batch normalization.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment