blob: 6a4dcc4542048df05219c44da1f69b9b30557f3b [file] [log] [blame]
graph(%a : Tensor) {
%1 : Device = prim::Constant[value="cpu"]()
%2 : int = prim::Constant[value=6]()
%i.1 : int = prim::Constant[value=0]()
%4 : int = prim::Constant[value=2]()
%5 : int = prim::Constant[value=3]()
%6 : int = prim::Constant[value=9223372036854775807]()
%7 : int = prim::Constant[value=1]()
%l : Tensor[] = prim::ListConstruct()
%9 : Tensor[] = aten::append(%l, %a)
%10 : int[] = prim::ListConstruct(%4, %5)
%b : Tensor = aten::rand(%10, %2, %i.1, %1)
%12 : bool = aten::lt(%i.1, %7)
%i : int = prim::Loop(%6, %12, %i.1)
block0(%14 : int, %15 : int) {
%c.1 : Tensor = aten::select(%l, %i.1)
%17 : int[] = prim::ListConstruct(%4, %5)
%18 : Tensor = aten::rand(%17, %2, %i.1, %1)
%c : Tensor = aten::add_(%c.1, %18, %7)
%i.2 : int = aten::add(%15, %7)
%21 : bool = aten::lt(%i.2, %7)
-> (%21, %i.2)
}
return (%b);
}