C++ API parity: MultiheadAttention
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27309
Test Plan: Imported from OSS
Differential Revision: D17766736
Pulled By: pbelevich
fbshipit-source-id: 7a5f2399f081945d31d4c13d7a8d248c387fc1a6