[moco-onnx] Use link helper (#6209)
author박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Mon, 5 Aug 2019 08:13:42 +0000 (17:13 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 5 Aug 2019 08:13:42 +0000 (17:13 +0900)
Let's use link helper instead of deprecated node method.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
compiler/moco-onnx/src/Frontend.cpp

index b671529..d633c1c 100644 (file)
@@ -224,7 +224,7 @@ void convert_graph(::onnx::ModelProto &onnx_model_proto, loco::Graph *graph)
     assert(pull_node != nullptr);
 
     graph_input->name(input);
-    graph_input->node(pull_node);
+    loco::link(graph_input, pull_node);
   }
 
   // 6. Create loco::Push node (with a proper input), and mark it as a graph output
@@ -242,7 +242,7 @@ void convert_graph(::onnx::ModelProto &onnx_model_proto, loco::Graph *graph)
     // set the graph output name and node object
     auto graph_output = graph->outputs()->create();
     graph_output->name(output);
-    graph_output->node(push_node);
+    loco::link(graph_output, push_node);
   }
 }