()
| 385 | |
| 386 | if attn_length is not None: |
| 387 | def attn_rnn_cell(): |
| 388 | return contrib_rnn.AttentionCellWrapper( |
| 389 | rnn_cell(), |
| 390 | attn_length=attn_length, |
| 391 | attn_size=attn_size, |
| 392 | attn_vec_size=attn_vec_size, |
| 393 | state_is_tuple=False) |
| 394 | else: |
| 395 | attn_rnn_cell = rnn_cell |
| 396 |