| graph(%0 : Double(2, 2) | |
| %1 : Double(2, 2) | |
| -------- stage 1 -------- | |
| %3 : Double(2, 2) | |
| %4 : Double(2, 2)) { | |
| %2 : Double(2, 2) = aten::add[alpha={1}](%0, %1) | |
| ---------------- stage 1 ---------------- | |
| %5 : Double(2, 2) = aten::add[alpha={1}](%3, %4) | |
| return (%2, %2, %5, %5); | |
| } |