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

Switch from contrib to non-contrib RNN cell

Browse Source 
PiperOrigin-RevId: 267532783
This commit is contained in:
Andrew Selle 2019-09-05 23:11:09 -07:00 committed by TensorFlower Gardener
parent ccd4d13357
commit cf27972e28
1 changed files with 1 additions and 1 deletions

2
tensorflow/lite/testing/generate_examples_lib.py
View File

@ -3484,7 +3484,7 @@ def make_lstm_tests(options):
# forget_bias == 0, inner state activation == tanh.
# TODO(zhixianyan): Add another test with forget_bias == 1.
# TODO(zhixianyan): Add another test with relu as activation.
lstm_cell = tf.contrib.rnn.BasicLSTMCell(
lstm_cell = tf.nn.rnn_cell.BasicLSTMCell(
num_cells, forget_bias=0.0, state_is_tuple=True)
cell_outputs, _ = rnn.static_rnn(
lstm_cell, inputs_after_split, dtype=tf.float32)