src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTDeepLabV3/call
class TFMobileViTDeepLabV3:    def call(self, hidden_states: tf.Tensor, training: bool = False) -> tf.Tensor:
        features = self.aspp(hidden_states[-1], training=training)
        features = self.dropout(features, training=training)
        features = self.classifier(features, training=training)
        return features