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

Remove unneeded casting of int64 for reverse_sequence (#12192)

Browse Source 
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 <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2017-08-10 17:19:11 -07:00 committed by Rasmus Munk Larsen
parent e1e81d9ba9
commit 35e7a36658
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/ops/rnn.py
View File

@ -294,10 +294,6 @@ def _reverse_seq(input_seq, lengths):
# Join into (time, batch_size, depth) # Join into (time, batch_size, depth)
s_joined = array_ops.stack(sequence) 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 # Reverse along dimension 0
s_reversed = array_ops.reverse_sequence(s_joined, lengths, 0, 1) s_reversed = array_ops.reverse_sequence(s_joined, lengths, 0, 1)
# Split again into list # Split again into list