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

Fix RNN benchmark code.

Browse Source 
PiperOrigin-RevId: 237279564
This commit is contained in:
Eugene Brevdo 2019-03-07 10:54:05 -08:00 committed by TensorFlower Gardener
parent 06404d88af
commit 4d1c0202aa
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/kernel_tests/rnn_test.py
View File

@ -723,7 +723,7 @@ def _static_vs_dynamic_rnn_benchmark_static(inputs_list_t, sequence_length):
use_peepholes=True, use_peepholes=True,
initializer=initializer, initializer=initializer,
state_is_tuple=False) state_is_tuple=False)
outputs, final_state = rnn_cell_impl.static_rnn( outputs, final_state = rnn.static_rnn(
cell, cell,
inputs_list_t, inputs_list_t,
sequence_length=sequence_length, sequence_length=sequence_length,
@ -856,7 +856,7 @@ def _half_seq_len_vs_unroll_half_rnn_benchmark(inputs_list_t, sequence_length):
use_peepholes=True, use_peepholes=True,
initializer=initializer, initializer=initializer,
state_is_tuple=False) state_is_tuple=False)
outputs, final_state = rnn_cell_impl.static_rnn( outputs, final_state = rnn.static_rnn(
cell, cell,
inputs_list_t, inputs_list_t,
sequence_length=sequence_length, sequence_length=sequence_length,
@ -922,7 +922,7 @@ def _concat_state_vs_tuple_state_rnn_benchmark(inputs_list_t, sequence_length,
use_peepholes=True, use_peepholes=True,
initializer=initializer, initializer=initializer,
state_is_tuple=state_is_tuple) state_is_tuple=state_is_tuple)
outputs, final_state = rnn_cell_impl.static_rnn( outputs, final_state = rnn.static_rnn(
cell, cell,
inputs_list_t, inputs_list_t,
sequence_length=sequence_length, sequence_length=sequence_length,