src/transformers/models/squeezebert/modeling_squeezebert.py/SqueezeBertLayerNorm/forward
class SqueezeBertLayerNorm:    def forward(self, x):
        x = x.permute(0, 2, 1)
        x = nn.LayerNorm.forward(self, x)
        return x.permute(0, 2, 1)