Cleanup the output descriptor of node arguments (#639)

64a105db · Jayaram Bobba · Scott Cyphers · d3ea93e2 · 64a105db
Commit 64a105db authored Mar 13, 2018 by Jayaram Bobba Committed by Scott Cyphers Mar 13, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

node.hpp src/ngraph/node.hpp +4 -0

No files found.
--- a/src/ngraph/node.hpp
+++ b/src/ngraph/node.hpp
@@ -71,6 +71,10 @@ namespace ngraph
            {
                arg->m_users.erase(this);
            }
+            for (auto& input : m_inputs)
+            {
+                input.get_output().remove_input(&input);
+            }
        }
        virtual void generate_adjoints(autodiff::Adjoints& adjoints,
                                       const std::shared_ptr<Node>& delta)