remove a node from users (#379)

* remove a node from users * style

remove a node from users (#379)
* remove a node from users * style
981dabef · varun-intel · Scott Cyphers · 5ad1de22 · 981dabef
Commit 981dabef authored Jan 17, 2018 by varun-intel Committed by Scott Cyphers Jan 17, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

node.hpp src/ngraph/node.hpp +7 -1

No files found.
--- a/src/ngraph/node.hpp
+++ b/src/ngraph/node.hpp
@@ -50,7 +50,13 @@ namespace ngraph

    protected:
        Node(const std::string& node_type, const Nodes& arguments);
-        virtual ~Node() {}
+        virtual ~Node()
+        {
+            for (auto arg : m_arguments)
+            {
+                arg->m_users.erase(this);
+            }
+        }
        virtual void generate_adjoints(autodiff::Adjoints& adjoints,
                                       const std::shared_ptr<Node>& delta)
        {