From 35e7a36658102f25c9e51b435c7a53cf21a2000a Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Thu, 10 Aug 2017 17:19:11 -0700 Subject: [PATCH] Remove unneeded casting of int64 for reverse_sequence (#12192) This fix remove unneeded cast of int64 for reverse_sequence: ``` lengths = math_ops.to_int64(lengths) ``` as int32 has already been enabled for reverse_sequence. Signed-off-by: Yong Tang --- tensorflow/python/ops/rnn.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tensorflow/python/ops/rnn.py b/tensorflow/python/ops/rnn.py index 0af72231b00..3c225abbbfc 100644 --- a/tensorflow/python/ops/rnn.py +++ b/tensorflow/python/ops/rnn.py @@ -294,10 +294,6 @@ def _reverse_seq(input_seq, lengths): # Join into (time, batch_size, depth) s_joined = array_ops.stack(sequence) - # TODO(schuster, ebrevdo): Remove cast when reverse_sequence takes int32 - if lengths is not None: - lengths = math_ops.to_int64(lengths) - # Reverse along dimension 0 s_reversed = array_ops.reverse_sequence(s_joined, lengths, 0, 1) # Split again into list