fix a bug in get_output_element (#1653)

02fad9e7 · Matthew Brookhart · Scott Cyphers · b0b8719d · 02fad9e7
Commit 02fad9e7 authored Sep 21, 2018 by Matthew Brookhart Committed by Scott Cyphers Sep 21, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

get_output_element.hpp src/ngraph/op/get_output_element.hpp +1 -1

No files found.
--- a/src/ngraph/op/get_output_element.hpp
+++ b/src/ngraph/op/get_output_element.hpp
@@ -50,7 +50,7 @@ namespace ngraph
    inline std::shared_ptr<Node> get_output_element(const std::shared_ptr<Node> node, size_t i = 0)
    {
-        return ((i == 0) && node->get_input_size() == 1)
+        return ((i == 0) && node->get_output_size() == 1)
                   ? node
                   : std::make_shared<op::GetOutputElement>(node, i);
    }