src/transformers/models/mobilebert/modeling_mobilebert.py/FFNOutput/forward
class FFNOutput:    def forward(self, hidden_states, residual_tensor):
        layer_outputs = self.dense(hidden_states)
        layer_outputs = self.LayerNorm(layer_outputs + residual_tensor)
        return layer_outputs