Searched refs:decoder_input_shape (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_nn.py | 3539 def test(encoder_input_shape, decoder_input_shape, argument 3547 decoder_input = torch.randn(decoder_input_shape) 3604 decoder_input_shape = correct_decoder_input_shape 3605 test(encoder_input_shape, decoder_input_shape) 3609 decoder_input_shape = update_shape(correct_decoder_input_shape, 1, wrong_bsz) 3610 test(encoder_input_shape, decoder_input_shape) 3614 decoder_input_shape = correct_decoder_input_shape 3615 test(encoder_input_shape, decoder_input_shape) 3619 decoder_input_shape = update_shape(correct_decoder_input_shape, 2, wrong_d_model) 3620 test(encoder_input_shape, decoder_input_shape) [all …]
|