src/transformers/models/resnet/modeling_tf_resnet.py/TFResNetForImageClassification/classifier
class TFResNetForImageClassification:    def classifier(self, x: tf.Tensor) -> tf.Tensor:
        x = tf.keras.layers.Flatten()(x)
        logits = self.classifier_layer(x)
        return logits