src/transformers/models/vit/modeling_flax_vit.py/FlaxViTSelfOutput/__call__
class FlaxViTSelfOutput:    def __call__(self, hidden_states, input_tensor, deterministic: bool = True):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        return hidden_states