src/transformers/models/beit/modeling_flax_beit.py/FlaxBeitIntermediate/__call__
class FlaxBeitIntermediate:    def __call__(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.activation(hidden_states)

        return hidden_states