Commit 453a6a3c authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball

make GOE extend from util::Op (#2153)

* make GOE extend from util::Op

* fix build breaks
parent a16c4961
...@@ -26,7 +26,7 @@ void regclass_pyngraph_op_GetOutputElement(py::module m) ...@@ -26,7 +26,7 @@ void regclass_pyngraph_op_GetOutputElement(py::module m)
{ {
py::class_<ngraph::op::GetOutputElement, py::class_<ngraph::op::GetOutputElement,
std::shared_ptr<ngraph::op::GetOutputElement>, std::shared_ptr<ngraph::op::GetOutputElement>,
ngraph::Node> ngraph::op::Op>
get_output_element(m, "GetOutputElement"); get_output_element(m, "GetOutputElement");
get_output_element.doc() = "ngraph.impl.op.GetOutputElement wraps ngraph::op::GetOutputElement"; get_output_element.doc() = "ngraph.impl.op.GetOutputElement wraps ngraph::op::GetOutputElement";
get_output_element.def(py::init<const std::shared_ptr<ngraph::Node>&, size_t>()); get_output_element.def(py::init<const std::shared_ptr<ngraph::Node>&, size_t>());
......
...@@ -22,7 +22,7 @@ using namespace std; ...@@ -22,7 +22,7 @@ using namespace std;
using namespace ngraph; using namespace ngraph;
op::GetOutputElement::GetOutputElement(const shared_ptr<Node>& arg, size_t n) op::GetOutputElement::GetOutputElement(const shared_ptr<Node>& arg, size_t n)
: Node("GetOutputElement", {arg}) : Op("GetOutputElement", {arg})
, m_n{n} , m_n{n}
{ {
constructor_validate_and_infer_types(); constructor_validate_and_infer_types();
......
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
#pragma once #pragma once
#include "ngraph/node.hpp" #include "ngraph/op/op.hpp"
namespace ngraph namespace ngraph
{ {
...@@ -25,7 +25,7 @@ namespace ngraph ...@@ -25,7 +25,7 @@ namespace ngraph
NodeVector get_output_elements(const std::shared_ptr<Node>& mon); NodeVector get_output_elements(const std::shared_ptr<Node>& mon);
/// \brief Operation to get an output from a node. /// \brief Operation to get an output from a node.
class GetOutputElement : public Node class GetOutputElement : public Op
{ {
public: public:
/// \brief Constructs a get-tuple-element operation. /// \brief Constructs a get-tuple-element operation.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment