megatron.model.bert_model.bert_extended_attention_mask# megatron.model.bert_model.bert_extended_attention_mask(attention_mask)#