blob: 422bcf88584b094bbbb10af2d2661a2985590c78 [file] [log] [blame]
graph(%x : Dynamic
%1 : Dynamic
%2 : Dynamic
%3 : Dynamic
%4 : Dynamic) {
%5 : int = prim::Constant[value=2]()
%6 : int = prim::Constant[value=1]()
%7 : int = prim::Constant[value=3]()
%8 : int = prim::Constant[value=27]()
%9 : Dynamic = aten::mul(%x, %x)
%10 : Dynamic = aten::add(%9, %7, %6)
%11 : int = aten::dim(%x)
%12 : bool = aten::eq(%11, %5)
%13 : bool = prim::If(%12)
block0() {
%14 : None = prim::None()
%15 : bool = aten::__isnot__(%2, %14)
-> (%15)
}
block1() {
-> (%12)
}
%ret.2 : Dynamic = prim::If(%13)
block0() {
%17 : Dynamic = aten::_unwrap_optional(%2)
%18 : Dynamic = aten::t(%1)
%ret.3 : Dynamic = aten::addmm(%17, %x, %18, %6, %6)
-> (%ret.3)
}
block1() {
%20 : Dynamic = aten::t(%1)
%output.3 : Dynamic = aten::matmul(%x, %20)
%22 : None = prim::None()
%23 : bool = aten::__isnot__(%2, %22)
%output.4 : Dynamic = prim::If(%23)
block0() {
%25 : Dynamic = aten::_unwrap_optional(%2)
%output.5 : Dynamic = aten::add_(%output.3, %25, %6)
-> (%output.5)
}
block1() {
-> (%output.3)
}
-> (%output.4)
}
%27 : Dynamic = aten::add(%10, %ret.2, %6)
%28 : Dynamic = aten::add(%x, %27, %6)
%29 : Dynamic = aten::add(%x, %8, %6)
%30 : Dynamic = aten::add(%28, %29, %6)
%31 : int = aten::dim(%x)
%32 : bool = aten::eq(%31, %5)
%33 : bool = prim::If(%32)
block0() {
%34 : None = prim::None()
%35 : bool = aten::__isnot__(%4, %34)
-> (%35)
}
block1() {
-> (%32)
}
%ret : Dynamic = prim::If(%33)
block0() {
%37 : Dynamic = aten::_unwrap_optional(%4)
%38 : Dynamic = aten::t(%3)
%ret.1 : Dynamic = aten::addmm(%37, %x, %38, %6, %6)
-> (%ret.1)
}
block1() {
%40 : Dynamic = aten::t(%3)
%output.1 : Dynamic = aten::matmul(%x, %40)
%42 : None = prim::None()
%43 : bool = aten::__isnot__(%4, %42)
%output : Dynamic = prim::If(%43)
block0() {
%45 : Dynamic = aten::_unwrap_optional(%4)
%output.2 : Dynamic = aten::add_(%output.1, %45, %6)
-> (%output.2)
}
block1() {
-> (%output.1)
}
-> (%output)
}
%47 : Dynamic = aten::add(%30, %ret, %6)
%48 : Dynamic = aten::add(%x, %47, %6)
return (%48);
}