src/fairseq/fairseq/model_parallel/modules/multihead_attention.py
src/fairseq/fairseq/modules/multihead_attention.py

src/fairseq/fairseq/model_parallel/modules/transformer_sentence_encoder.py
src/fairseq/fairseq/modules/transformer_sentence_encoder.py

src/fairseq/fairseq/model_parallel/modules/transformer_sentence_encoder_layer.py
src/fairseq/fairseq/modules/transformer_sentence_encoder_layer.py

src/fairseq/fairseq/model_parallel/modules/multihead_attention.py
src/fairseq/fairseq/modules/longformer_multihead_attention.py