src/transformers/models/perceiver/modeling_perceiver.py/Conv2DDownsample/forward
class Conv2DDownsample:    def forward(self, inputs: torch.Tensor) -> torch.Tensor:
        out = self.conv(inputs)
        out = self.batchnorm(out)
        out = self.relu(out)
        out = self.max_pool(out)
        return out