blob: 4efe821cefbae1c3eb3ae49776170ee311caa14c [file] [log] [blame]
graph(%a : Tensor) {
%1 : int = prim::Constant[value=2]()
%2 : Tensor = aten::lt(%a, %1)
%3 : bool = prim::TensorToBool(%2)
%c : int = prim::If(%3)
block0() {
%5 : int = prim::Constant[value=5]()
-> (%5)
}
block1() {
%6 : int = prim::Constant[value=1]()
-> (%6)
}
return (%c);
}