Home
last modified time | relevance | path

Searched refs:decoder_input_shape (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py3539 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 …]