src/transformers/models/efficientformer/modeling_efficientformer.py/EfficientFormerConvStem/forward
class EfficientFormerConvStem:    def forward(self, pixel_values: torch.Tensor) -> torch.Tensor:
        features = self.batchnorm_before(self.convolution1(pixel_values))
        features = self.activation(features)
        features = self.batchnorm_after(self.convolution2(features))
        features = self.activation(features)

        return features