src/transformers/models/beit/modeling_flax_beit.py/FlaxBeitOutput/__call__
class FlaxBeitOutput:    def __call__(self, hidden_states, deterministic: bool = True):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)

        return hidden_states