def METHOD_NAME(self, hidden_states, attention_mask,
                               encoder_hidden_states,
                               encoder_attention_mask):
    cat_hidden_states = torch.cat([hidden_states, encoder_hidden_states],
                                  dim=1)
    cat_attention_mask = torch.cat(
        [attention_mask, encoder_attention_mask], dim=1)
    cat_attention_mask = self.plm.get_extended_attention_mask(
        cat_attention_mask,
        cat_hidden_states.size()[:2])
    hidden_states = self.crossattention(
        hidden_states=cat_hidden_states, attention_mask=cat_attention_mask
    )[0][:, :hidden_states.size()[1], :]
    return hidden_states