def METHOD_NAME(self, x: Tensor) -> Tensor:
    """Forward function for SimplifiedBasicBlock."""
    identity = x
    out = self.conv1(x)
    if self.with_norm:
        out = self.norm1(out)
    out = self.relu(out)
    out = self.conv2(out)
    if self.with_norm:
        out = self.norm2(out)
    if self.downsample is not None:
        identity = self.downsample(x)
    out += identity
    return out