blob: 276fadc61fd7dfe2631c1cf62f9320f0b03f8642 [file] [log] [blame]
graph(%0 : Float(1)
%1 : Float(1)) {
%2 : Float(1) = prim::FusionGroup_0[device=1](%0, %1)
return (%2);
}
with prim::FusionGroup_0 = graph(%6 : Float(1)
%9 : Float(1)) {
%10 : Float(1) = aten::add[alpha={1}](%6, %9)
%8 : Float(1) = aten::mul(%6, %10)
%5 : Float(1) = aten::add[other={1}, alpha={1}](%8)
%3 : Float(1) = aten::tanh(%5)
%1 : Float(1) = aten::sigmoid(%3)
return (%1);
}