src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTOutput/__init__
class TFMobileViTOutput:    def __init__(self, config: MobileViTConfig, hidden_size: int, intermediate_size: int, **kwargs) -> None:
        super().__init__(**kwargs)
        self.dense = tf.keras.layers.Dense(hidden_size, name="dense")
        self.dropout = tf.keras.layers.Dropout(config.hidden_dropout_prob)