[locomotiv] Remove GraphInput/Output node method calls (#6719)

This commit removes all the GraphInput::node and GraphOutput::node
method calls in locomotiv. This is a step toward loco GraphInput/Output
class redesign.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/compiler/locomotiv/src/Session.cpp b/compiler/locomotiv/src/Session.cpp
index bff632d..258603d 100644 (file)
--- a/compiler/locomotiv/src/Session.cpp
+++ b/compiler/locomotiv/src/Session.cpp
@@ -41,7 +41,7 @@ void Session::set_input(uint32_t index, std::unique_ptr<NodeData> &&data)
   assert(index < input_size());
 
   // Check whether already annotated
-  auto pull = _graph->inputs()->at(index)->node();
+  auto pull = loco::pull_node(_graph, index);
   if (annot_data(pull))
   {
     throw std::runtime_error("Graph input already has NodeData");

diff --git a/compiler/locomotiv/src/Session.test.cpp b/compiler/locomotiv/src/Session.test.cpp
index 07d55d2..59fb9b7 100644 (file)
--- a/compiler/locomotiv/src/Session.test.cpp
+++ b/compiler/locomotiv/src/Session.test.cpp
@@ -360,14 +360,14 @@ TEST(Session, dtor)
 
     s.set_input(0, std::move(data));
 
-    auto data_annotated = locomotiv::annot_data(input->node());
+    auto data_annotated = locomotiv::annot_data(pull);
     ASSERT_NE(data_annotated, nullptr);
-    auto domain_annotated = locomotiv::annot_domain(input->node());
+    auto domain_annotated = locomotiv::annot_domain(pull);
     ASSERT_EQ(domain_annotated, loco::Domain::Tensor);
   }
 
-  auto data_annotated = locomotiv::annot_data(input->node());
+  auto data_annotated = locomotiv::annot_data(pull);
   ASSERT_EQ(data_annotated, nullptr);
-  auto domain_annotated = locomotiv::annot_domain(input->node());
+  auto domain_annotated = locomotiv::annot_domain(pull);
   ASSERT_EQ(domain_annotated, loco::Domain::Unknown);
 }