pytorch
b9afa928 - replace double transpose with single permute in nn.f.mha (#89847)

Commit
2 years ago
replace double transpose with single permute in nn.f.mha (#89847) # Summary I forgot about permute which was exactly what I wanted. Quick perf bump Pull Request resolved: https://github.com/pytorch/pytorch/pull/89847 Approved by: https://github.com/cpuhrsch, https://github.com/albanD
Author
Committer
Parents
Loading