src/transformers/models/mobilebert/modeling_mobilebert.py/BottleneckLayer/forward
class BottleneckLayer:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        layer_input = self.dense(hidden_states)
        layer_input = self.LayerNorm(layer_input)
        return layer_input