diff options
Diffstat (limited to 'tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py')
-rw-r--r-- | tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py b/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py index 815461788a..7b85f998ea 100644 --- a/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py +++ b/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py @@ -1329,6 +1329,7 @@ def batch_sequences_with_states(input_key, input_key=key, input_sequences=sequences, input_context=context, + input_length=tf.shape(sequences["input"])[0], initial_states=initial_states, num_unroll=num_unroll, batch_size=batch_size, |