src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTASPPPooling/__init__
class TFMobileViTASPPPooling:    def __init__(self, config: MobileViTConfig, out_channels: int, **kwargs) -> None:
        super().__init__(**kwargs)

        self.global_pool = tf.keras.layers.GlobalAveragePooling2D(keepdims=True, name="global_pool")

        self.conv_1x1 = TFMobileViTConvLayer(
            config,
            out_channels=out_channels,
            kernel_size=1,
            stride=1,
            use_normalization=True,
            use_activation="relu",
            name="conv_1x1",
        )