| graph(%a : Tensor) { | |
| %1 : int = prim::Constant[value=2]() | |
| %2 : Tensor = aten::lt(%a, %1) | |
| %3 : bool = prim::TensorToBool(%2) | |
| %c : int = prim::If(%3) | |
| block0() { | |
| %5 : int = prim::Constant[value=5]() | |
| -> (%5) | |
| } | |
| block1() { | |
| %6 : int = prim::Constant[value=1]() | |
| -> (%6) | |
| } | |
| return (%c); | |
| } |