src/transformers/models/cvt/modeling_tf_cvt.py/TFCvtForImageClassification/serving_output
class TFCvtForImageClassification:    def serving_output(self, output: TFImageClassifierOutputWithNoAttention) -> TFImageClassifierOutputWithNoAttention:
        return TFImageClassifierOutputWithNoAttention(logits=output.logits, hidden_states=output.hidden_states)