src/transformers/models/mobilevit/modeling_mobilevit.py/MobileViTInvertedResidual/forward
class MobileViTInvertedResidual:    def forward(self, features: torch.Tensor) -> torch.Tensor:
        residual = features

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

        return residual + features if self.use_residual else features