pytorch
e0f2eba9 - Move odd num_head in TransformerEncoder to slow_path (#83483)

Commit
3 years ago
Move odd num_head in TransformerEncoder to slow_path (#83483) Summary: odd nhead is not supported for masked softmax, therefore we just move it to use old slow_path Test Plan: CI Differential Revision: D38720086 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83483 Approved by: https://github.com/erichan1
Author
Committer
Parents
Loading