graph(%x : Tensor %y : Tensor) { %2 : int = prim::Constant[value=1]() %3 : Tensor = aten::add(%x, %y, %2) %4 : Tensor = aten::gt(%3, %x) %5 : bool = prim::Bool(%4) %z : Tensor = prim::If(%5) block0() { -> (%3) } block1() { -> (%x) } return (%z); }