blob: 7f7340e1ff13124c429035fb6bd7d80bce739fd0 [file] [log] [blame]
graph(%cond : Tensor) {
%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 : Device = prim::Constant[value="cpu"]()
%12 : bool = prim::Bool(%cond)
%c : int, %y : Tensor = prim::If(%12)
block0() {
-> (%3, %4)
}
block1() {
%y.2 : Tensor = aten::rand(%2, %10, %c.1, %11)
%16 : bool = prim::Bool(%cond)
%y.4 : Tensor = prim::If(%16)
block0() {
%y.3 : Tensor = 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, %y);
}