using Output<Node> instead of shared_ptr<Node>

src/ngraph/op/recv.cpp

@@ -21,7 +21,7 @@ using namespace ngraph;
 
 const string op::Recv::type_name{"Recv"};
 
-op::Recv::Recv(const shared_ptr<Node>& arg, int src_id)
+op::Recv::Recv(const Output<Node>& arg, int src_id)
     : Op(check_single_output_args({arg}))
     , m_src_id(src_id)
 {

src/ngraph/op/recv.hpp

@@ -36,7 +36,7 @@ namespace ngraph
             ///
             /// \param arg The node for tensor to receive data
             /// \param src_id the source id which could be rank or node id.
-            Recv(const std::shared_ptr<Node>& arg, int src_id);
+            Recv(const Output<Node>& arg, int src_id);
 
             void validate_and_infer_types() override;
 

src/ngraph/op/send.cpp

@@ -21,7 +21,7 @@ using namespace ngraph;
 
 const string op::Send::type_name{"Send"};
 
-op::Send::Send(const shared_ptr<Node>& arg, int dest_id)
+op::Send::Send(const Output<Node>& arg, int dest_id)
     : Op(check_single_output_args({arg}))
     , m_dest_id(dest_id)
 {

src/ngraph/op/send.hpp

@@ -36,7 +36,7 @@ namespace ngraph
             ///
             /// \param arg The node for input tensor
             /// \param dest_id the target id which could be rank of node id.
-            Send(const std::shared_ptr<Node>& arg, int dest_id);
+            Send(const Output<Node>& arg, int dest_id);
 
             void validate_and_infer_types() override;