src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTInvertedResidual/call
class TFMobileViTInvertedResidual:    def call(self, features: tf.Tensor, training: bool = False) -> tf.Tensor:
        residual = features

        features = self.expand_1x1(features, training=training)
        features = self.conv_3x3(features, training=training)
        features = self.reduce_1x1(features, training=training)

        return residual + features if self.use_residual else features