src/transformers/models/resnet/modeling_tf_resnet.py/TFResNetConvLayer/call
class TFResNetConvLayer:    def call(self, hidden_state: tf.Tensor, training: bool = False) -> tf.Tensor:
        hidden_state = self.convolution(hidden_state)
        hidden_state = self.normalization(hidden_state, training=training)
        hidden_state = self.activation(hidden_state)
        return hidden_state