def METHOD_NAME(self) -> None:
    class M(torch.nn.Module):
        def __init__(self):
            super().__init__()
        def forward(self, pred, x, y):
            def true_fn(x, y):
                b = x.item()
                constrain_as_value(b, min=2, max=5)
                return x - y
            def false_fn(x, y):
                c = y.item()
                constrain_as_value(c, min=2, max=5)
                return x + y
            ret = control_flow.cond(pred, true_fn, false_fn, [x, y])
            return ret
    x = torch.tensor([2])
    y = torch.tensor([5])
    mod = M()
    _ = export(mod, (torch.tensor(True), x, y))._transform(_ExportPassBase())