rename to need_attn_weights to match elsewhere (#97102)
Change variable spelling from `need_atten_weights` to `need_attn_weights` to match naming convention elsewhere in pytorch.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97102
Approved by: https://github.com/drisspg
