From 23b131aac883707f4054b1fddc1927b732120542 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/On-Device=20Lab=28SR=29/Staff?= =?utf8?q?=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 20:02:54 +0900 Subject: [PATCH] [mir2loco] Use link helper (#6211) This commit revises mir2loco to use link helper function instead of deprecated node method in GraphInput and GraphOutput. Signed-off-by: Jonghyun Park --- compiler/mir2loco/src/mir2loco.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/compiler/mir2loco/src/mir2loco.cpp b/compiler/mir2loco/src/mir2loco.cpp index f236039..c005b30 100644 --- a/compiler/mir2loco/src/mir2loco.cpp +++ b/compiler/mir2loco/src/mir2loco.cpp @@ -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); } -- 2.7.4