blob: 7f6e68490b3502208ccbf3c8aff1684ac3c3bf7f [file] [log] [blame]
graph(%x : Float(*, *),
%y : Float(*, *)):
%2 : Tensor[] = prim::ListConstruct(%x, %y)
%3 : Tensor[] = aten::broadcast_tensors(%2)
%4 : Tensor, %5 : Tensor = prim::ListUnpack(%3)
%6 : Float(*, *) = prim::FusionGroup_0(%5, %4)
return (%6)
with prim::FusionGroup_0 = graph(%0 : Tensor,
%1 : Tensor):
%2 : Float(*, *), %3 : Float(*, *) = prim::ConstantChunk[chunks=2, dim=1](%1)
%4 : Float(*, *), %5 : Float(*, *) = prim::ConstantChunk[chunks=2, dim=1](%0)
%6 : int = prim::Constant[value=1]()
%7 : Float(*, *) = aten::add(%2, %4, %6)
%8 : Float(*, *) = aten::add(%3, %5, %6)
%9 : Float(*, *) = aten::mul(%7, %8)
return (%9)