()
| 31 | |
| 32 | |
| 33 | def init_decoder_model(): |
| 34 | return TransformerDecoder( |
| 35 | vocab_size=10, |
| 36 | encoder_output_size=40, |
| 37 | attention_heads=4, |
| 38 | linear_units=40, |
| 39 | num_blocks=2, |
| 40 | input_layer="embed", |
| 41 | ) |
| 42 | |
| 43 | |
| 44 | @pytest.mark.skipif( |
no test coverage detected
searching dependent graphs…