From 4d1c0202aa1cc080f2cb700fd2d4530db3b55d14 Mon Sep 17 00:00:00 2001 From: Eugene Brevdo Date: Thu, 7 Mar 2019 10:54:05 -0800 Subject: [PATCH] Fix RNN benchmark code. PiperOrigin-RevId: 237279564 --- tensorflow/python/kernel_tests/rnn_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/python/kernel_tests/rnn_test.py b/tensorflow/python/kernel_tests/rnn_test.py index 2ebb2db2224..bdb01700582 100644 --- a/tensorflow/python/kernel_tests/rnn_test.py +++ b/tensorflow/python/kernel_tests/rnn_test.py @@ -723,7 +723,7 @@ def _static_vs_dynamic_rnn_benchmark_static(inputs_list_t, sequence_length): use_peepholes=True, initializer=initializer, state_is_tuple=False) - outputs, final_state = rnn_cell_impl.static_rnn( + outputs, final_state = rnn.static_rnn( cell, inputs_list_t, 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, initializer=initializer, state_is_tuple=False) - outputs, final_state = rnn_cell_impl.static_rnn( + outputs, final_state = rnn.static_rnn( cell, inputs_list_t, sequence_length=sequence_length, @@ -922,7 +922,7 @@ def _concat_state_vs_tuple_state_rnn_benchmark(inputs_list_t, sequence_length, use_peepholes=True, initializer=initializer, state_is_tuple=state_is_tuple) - outputs, final_state = rnn_cell_impl.static_rnn( + outputs, final_state = rnn.static_rnn( cell, inputs_list_t, sequence_length=sequence_length,