blob: d01001c1ff1c7553642934133216c83974414f80 [file] [log] [blame]
graph(%cond : Dynamic) {
%1 : int[] = prim::Constant[value=[1]]()
%2 : int[] = prim::Constant[value=[0]]()
%3 : int = prim::Constant[value=3]()
%4 : Float(2) = prim::Constant[value= 4 4 [ CPUFloatType{2} ]]()
%5 : Float(2) = prim::Constant[value= 1 1 [ CPUFloatType{2} ]]()
%c.1 : int = prim::Constant[value=0]()
%a : int = prim::Constant[value=1]()
%d : string = prim::Constant[value="abc"]()
%e : string = prim::Constant[value="bcd"]()
%10 : int = prim::Constant[value=6]()
%11 : int[] = prim::Constant[value=[0, -1]]()
%12 : bool = prim::TensorToBool(%cond)
%c : int, %y : Dynamic = prim::If(%12)
block0() {
-> (%3, %4)
}
block1() {
%y.2 : Dynamic = aten::rand(%2, %10, %c.1, %11)
%16 : bool = prim::TensorToBool(%cond)
%y.4 : Dynamic = prim::If(%16)
block0() {
%y.3 : Dynamic = aten::rand(%1, %10, %c.1, %11)
-> (%y.3)
}
block1() {
-> (%y.2)
}
= prim::Print(%d, %e, %d, %5, %y.4, %5)
-> (%c.1, %y.4)
}
return (%a, %3, %c, %5);
}