fix

7 years ago · da5efa735a
parent 5daad16218
commit da5efa735a
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/details/var_handle.h
+++ b/paddle/fluid/framework/details/var_handle.h
@ -57,7 +57,8 @@ struct VarHandleBase {
  void RemoveOutput(OpHandleBase* out, ir::Node* node) {
    pending_ops_.erase(out);
    node_->outputs.erase(
-        std::remove(node_->outputs.begin(), node_->outputs.end(), node));
+        std::remove(node_->outputs.begin(), node_->outputs.end(), node),
+        node_->outputs.end());
  }

  void ClearGeneratedOp() {