Update RNN helpers to be able to handle dynamic state sizes.
This fixes a bug I introduced previously by adding the alignment into the AttentionWrapper's state (since the alignment's size may have to be a Tensor - the encoder's max_time is not usually static). PiperOrigin-RevId: 156077314
Loading
Please sign in to comment