transformers

fix type annotation for ALL_ATTENTION_FUNCTIONS

#36690

Merged