[mir2loco] Use link helper (#6211)
author박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Mon, 5 Aug 2019 11:02:54 +0000 (20:02 +0900)
committerEfimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Mon, 5 Aug 2019 11:02:54 +0000 (14:02 +0300)
This commit revises mir2loco to use link helper function instead of
deprecated node method in GraphInput and GraphOutput.

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

index f236039..c005b30 100644 (file)
@@ -230,7 +230,7 @@ void Transformer::visit(mir::ops::InputOp &op)
   auto graph_input = _loco_graph->inputs()->create();
   graph_input->name(op.getName());
   graph_input->dtype(loco::DataType::FLOAT32); // TODO Support other types
-  graph_input->node(pull_node);
+  loco::link(graph_input, pull_node);
   // Not set inputs
   // Add to map
   _mir2loco_map.emplace(&op, pull_node);
@@ -252,7 +252,7 @@ void Transformer::visit(mir::ops::OutputOp &op)
   auto graph_output = _loco_graph->outputs()->create();
   graph_output->name(op.getName());
   graph_output->dtype(loco::DataType::FLOAT32); // TODO Support other types
-  graph_output->node(push_node);
+  loco::link(graph_output, push_node);
   // Add to map
   _mir2loco_map.emplace(&op, push_node);
 }