src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertSelfOutput/forward
class MegatronBertSelfOutput:    def forward(self, hidden_states, residual):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states)
        return residual + hidden_states