src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTForImageClassification/serving_output
class TFMobileViTForImageClassification:    def serving_output(self, output: TFImageClassifierOutputWithNoAttention) -> TFImageClassifierOutputWithNoAttention:
        # hidden_states and attention not converted to Tensor with tf.convert_to_tensor as they are all of different dimensions
        return TFImageClassifierOutputWithNoAttention(logits=output.logits, hidden_states=output.hidden_states)