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