| graph(%x.1_data : Dynamic |
| %x.1_mask : Dynamic |
| %x.1_dims : Dynamic |
| %y_data : Dynamic |
| %y_mask : Dynamic |
| %y_dims : Dynamic) { |
| %6 : int = prim::Constant[value=1]() |
| %7 : bool = prim::Constant[value=1]() |
| %8 : int = prim::Constant[value=10]() |
| %x : Dynamic, %10 : Dynamic, %11 : Dynamic = prim::Loop(%8, %7, %x.1_data, %x.1_mask, %x.1_dims) |
| block0(%loop_num : int, %5_data : Dynamic, %5_mask : Dynamic, %5_dims : Dynamic) { |
| %16 : Long() = prim::NumToTensor(%6) |
| %alpha : float = prim::TensorToNum(%16) |
| %data.1 : Dynamic = aten::add(%5_data, %y_data, %alpha) |
| %mask : Dynamic = aten::mul(%5_mask, %y_mask) |
| %dims : Dynamic = aten::__or__(%5_dims, %y_dims) |
| %data : Dynamic = aten::where(%mask, %data.1, %5_data) |
| -> (%7, %data, %mask, %dims) |
| } |
| return (%x, %10, %11); |
| } |