1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
graph(%x.1_data : Tensor
%x.1_mask : Tensor
%x.1_dims : Tensor
%y_data : Tensor
%y_mask : Tensor
%y_dims : Tensor) {
%6 : int = prim::Constant[value=1]()
%7 : bool = prim::Constant[value=1]()
%8 : int = prim::Constant[value=10]()
%x : Tensor, %10 : Tensor, %11 : Tensor = prim::Loop(%8, %7, %x.1_data, %x.1_mask, %x.1_dims)
block0(%loop_num : int, %5_data : Tensor, %5_mask : Tensor, %5_dims : Tensor) {
%16 : Long() = prim::NumToTensor(%6)
%alpha : float = prim::Float(%16)
%data.1 : Tensor = aten::add(%5_data, %y_data, %alpha)
%mask : Tensor = aten::mul(%5_mask, %y_mask)
%dims : Tensor = aten::__or__(%5_dims, %y_dims)
%data : Tensor = aten::where(%mask, %data.1, %5_data)
-> (%7, %data, %mask, %dims)
}
%22 : (Tensor, Tensor, Tensor) = prim::TupleConstruct(%x, %10, %11)
return (%22);
}
|
