graph(%0 : Tensor | |
%1 : Tensor) { | |
%2 : int = prim::Constant[value=1]() | |
%3 : Double(1) = aten::add(%0, %1, %2) | |
%4 : Double(1) = aten::mul(%0, %3) | |
%5 : Double(1) = aten::tanh(%4) | |
%6 : Double(1) = aten::sigmoid(%5) | |
%7 : Tensor = prim::TensorTest[a= 1 1 1 1 [ CPUDoubleType{2,2} ]]() | |
return (%6); | |
} |