pytorch
7230b67c - [PyTorch] Support norm_first in nn.TransformerEncoderLayer fast path (#78269)

Commit
2 years ago
[PyTorch] Support norm_first in nn.TransformerEncoderLayer fast path (#78269) Straightforward after previous diffs in stack cleaning up code and adding test coverage. Differential Revision: [D36564008](https://our.internmc.facebook.com/intern/diff/D36564008/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/78269 Approved by: https://github.com/jbschlosser
Author
Committer
Parents
Loading