blob: a21d867f01e7790e1e41f375d9c7bd1c397bf5db [file] [log] [blame]
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);
}