[BugFix] Fix bug in onnx::SsaRewrite (pytorch#42148)

Summary: Pull Request resolved: pytorch#42148 Differential Revision: D22687388 fbshipit-source-id: facf7a186dd48d6f919d0ff5d42f756977c3f9f4
zmmwl · Jul 28, 2020 · 5336ccc · 5336ccc
1 parent 4f72382
commit 5336ccc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/caffe2/onnx/onnx_exporter.cc b/caffe2/onnx/onnx_exporter.cc
@@ -258,6 +258,8 @@ void revertRenamedExternalOutputForIfOp(
   ArgumentHelper helper(*if_op);
   Argument *then_arg = nullptr, *else_arg = nullptr;
 
+  revertRenamedExternalOutput(if_op, renamed_external_outputs);
+
   if (helper.HasSingleArgumentOfType<NetDef>("then_net")) {
     then_arg = getArgumentFromName(if_op, "then_net");
     NetDef* net = then_arg->mutable_n();