src/transformers/models/resnet/modeling_tf_resnet.py/TFResNetStage/call
class TFResNetStage:    def call(self, hidden_state: tf.Tensor, training: bool = False) -> tf.Tensor:
        for layer in self.stage_layers:
            hidden_state = layer(hidden_state, training=training)
        return hidden_state