Modify dynamic_rnn to use max_seq_length for its while_loop.
This is especially important with XLA compilation, which will usually have a maximum time dimension on the input that is (sometimes much larger than) max(sequence_length). In these cases, a lot of compute can be saved. PiperOrigin-RevId: 183148831
Loading
Please sign in to comment