()
| 356 | state_is_tuple=False) |
| 357 | |
| 358 | def attn_bw_cell(): |
| 359 | return contrib_rnn.AttentionCellWrapper( |
| 360 | bw_cell(), |
| 361 | attn_length=attn_length, |
| 362 | attn_size=attn_size, |
| 363 | attn_vec_size=attn_vec_size, |
| 364 | state_is_tuple=False) |
| 365 | else: |
| 366 | attn_fw_cell = fw_cell |
| 367 | attn_bw_cell = bw_cell |