src/transformers/models/cvt/modeling_tf_cvt.py/TFCvtOutput/__init__
class TFCvtOutput:    def __init__(self, config: CvtConfig, embed_dim: int, drop_rate: int, **kwargs):
        super().__init__(**kwargs)
        self.dense = tf.keras.layers.Dense(
            units=embed_dim, kernel_initializer=get_initializer(config.initializer_range), name="dense"
        )
        self.dropout = tf.keras.layers.Dropout(drop_rate)