def METHOD_NAME(self, x):
    # Inference mode forward pass.
    if hasattr(self, "reparam_conv"):
        return self.reparam_conv(x)
    # Multi-branched train-time forward pass.
    # Skip branch output
    identity_out = 0
    if self.rbr_skip is not None:
        identity_out = self.rbr_skip(x)
    # Scale branch output
    scale_out = 0
    if self.rbr_scale is not None:
        scale_out = self.rbr_scale(x)
    # Other branches
    out = scale_out + identity_out
    out += self.rbr_conv(x)
    return out