- addressing github issue #1198 (reverse_sequence only accepts int64)

Change: 116671488
tensorflow · Mar 9, 2016 · 484a80c · 484a80c
1 parent 439a269
commit 484a80c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/python/ops/rnn.py b/tensorflow/python/ops/rnn.py
@@ -269,6 +269,10 @@ def _reverse_seq(input_seq, lengths):
   # Join into (time, batch_size, depth)
   s_joined = array_ops.pack(input_seq)
 
+  # 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