Michael Gschwind 36e1ae6778 De-select odd numbered heads from nn.MHA fastpath (#99672) ... Summary: https://github.com/pytorch/pytorch/issues/97128 * Add test for mha num_heads %2 != 0 * Fix test * Add test for bias false * show test passes Test Plan: sandcastle Differential Revision: D45161767 Pull Request resolved: https://github.com/pytorch/pytorch/pull/99672 Approved by: https://github.com/ngimel		2023-04-25 00:27:18 +00:00
..
test_convolution.py
test_dropout.py
test_embedding.py
test_init.py
test_lazy_modules.py
test_module_hooks.py	Fix module backward pre-hooks to actually update gradient (#97983)	2023-03-30 20:33:44 +00:00
test_multihead_attention.py	De-select odd numbered heads from nn.MHA fastpath (#99672)	2023-04-25 00:27:18 +00:00
test_packed_sequence.py	Move empty check to the start of _pack_padded_sequence (#94885)	2023-03-22 04:16:58 +00:00
test_parametrization.py
test_pooling.py
test_pruning.py