Commit 275e7d02 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by Amit Patankar
Browse files

[tf contrib seq2seq] Improve BeamSearchDecoder's ability to handle unknown shapes. 

Updated unit tests to contain inputs of unknown shape (at graph build time).
Found an issue in the gather helper that stops it from properly propagating
the batch size of the output shape.  This caused problems with tf.while_loop.
Fixed.

PiperOrigin-RevId: 157533937
parent 93e5eaad
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment