pytorch
4fada6eb - MHA torch.jit.script fix for in_proj_weight = None (#95653)

Commit
2 years ago
MHA torch.jit.script fix for in_proj_weight = None (#95653) Summary: MHA fix to support in_proj_weight being None Test Plan: sandcastle Differential Revision: D43628206 Pull Request resolved: https://github.com/pytorch/pytorch/pull/95653 Approved by: https://github.com/davidberard98, https://github.com/cpuhrsch
Author
Michael Gschwind
Committer
Parents
Loading