src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTSelfOutput/call
class TFMobileViTSelfOutput:    def call(self, hidden_states: tf.Tensor, training: bool = False) -> tf.Tensor:
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states, training=training)
        return hidden_states