Make mocked functioned return the proper result structure (tuple for native MHA for attn result and attn weights) (#106526)
Summary: Make mocked functioned return the proper result structure (tuple for native MHA for attn result and attn weights)
Test Plan: sandcastle
Differential Revision: D48021277
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106526
Approved by: https://github.com/davidberard98
