def METHOD_NAME(self, x):
    x_shape = x.shape
    x = x.reshape(-1, x.size(-1))
    x = self.fc1(x)
    x = self.activation_fn(x.float()).type_as(x)
    x = self.activation_dropout_module(x)
    if self.ffn_layernorm is not None:
        x = self.ffn_layernorm(x)
    x = self.fc2(x)
    x = x.view(x_shape)
    x = self.dropout_module(x)
    return x