fix assert (#1660)

e0384a8d · Nick Korovaiko · Robert Kimball · ad2e0097 · e0384a8d
Commit e0384a8d authored Sep 21, 2018 by Nick Korovaiko Committed by Robert Kimball Sep 21, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

node.cpp src/ngraph/node.cpp +2 -4

No files found.
--- a/src/ngraph/node.cpp
+++ b/src/ngraph/node.cpp
@@ -154,10 +154,8 @@ std::shared_ptr<Node> Node::get_argument(size_t index) const
 {
    for (auto& i : get_inputs())
    {
-        if (i.get_output().get_node()->get_outputs().size() != 1)
-        {
-            throw "get_argument called on an argument w/ multiple outputs";
-        }
+        NGRAPH_ASSERT(i.get_output().get_node()->get_outputs().size() == 1)
+            << "child " << i.get_output().get_node()->get_name() << " has multiple outputs";
    }
    return m_inputs.at(index).get_output().get_node();
 }