From 31c1f1daf30d1384df302068439cbc75542efdfe 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 15:03:25 +0900 Subject: [PATCH] [exo-tflite] Use link helper (#6200) This commit revises exo-tflite to use link helper instead of deprecated "node" method in GraphInput/GraphOutput. Signed-off-by: Jonghyun Park --- compiler/exo-tflite/src/TFLExporterImpl.test.cpp | 8 ++++---- compiler/exo-tflite/src/TypeInference.test.cpp | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/compiler/exo-tflite/src/TFLExporterImpl.test.cpp b/compiler/exo-tflite/src/TFLExporterImpl.test.cpp index 3d9b921..d1b8217 100644 --- a/compiler/exo-tflite/src/TFLExporterImpl.test.cpp +++ b/compiler/exo-tflite/src/TFLExporterImpl.test.cpp @@ -97,12 +97,12 @@ TEST_F(TFLExporterImplTests, Relu6) auto input = graph()->inputs()->create(); { input->name("input"); - input->node(pull); + loco::link(input, pull); } auto output = graph()->outputs()->create(); { output->name("output"); - output->node(push); + loco::link(output, push); } exo::TFLExporter::Impl exporter{graph()}; @@ -141,12 +141,12 @@ TEST_F(TFLExporterImplTests, Regression_0000) auto input = graph()->inputs()->create(); { input->name("input"); - input->node(pull); + loco::link(input, pull); } auto output = graph()->outputs()->create(); { output->name("output"); - output->node(push); + loco::link(output, push); } exo::TFLExporter::Impl exporter{graph()}; diff --git a/compiler/exo-tflite/src/TypeInference.test.cpp b/compiler/exo-tflite/src/TypeInference.test.cpp index 3a922e8..10cad8f 100644 --- a/compiler/exo-tflite/src/TypeInference.test.cpp +++ b/compiler/exo-tflite/src/TypeInference.test.cpp @@ -35,7 +35,7 @@ public: auto graph_input = _graph.inputs()->create(); graph_input->name("graph_input"); - graph_input->node(pull); + loco::link(graph_input, pull); pull->dtype(dtype); setSampleShape(pull); @@ -58,7 +58,7 @@ public: auto graph_output = _graph.outputs()->create(); graph_output->name("graph_output"); - graph_output->node(push); + loco::link(graph_output, push); push->from(_last); -- 2.7.4