[neurun] Add const iterate method to Linear (#3547)

Add const iterate method to Linear

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/runtimes/neurun/src/linear/Linear.cc b/runtimes/neurun/src/linear/Linear.cc
index 59a0050..2f97731 100644 (file)
--- a/runtimes/neurun/src/linear/Linear.cc
+++ b/runtimes/neurun/src/linear/Linear.cc
@@ -184,5 +184,13 @@ backend::TensorBuilderSet Linear::planTensors()
   return tensor_builders;
 }
 
+void Linear::iterate(const std::function<void(const graph::operation::Node *)> &fn) const
+{
+  for (const auto op : _operations)
+  {
+    fn(op);
+  }
+}
+
 } // namespace linear
 } // namespace neurun

diff --git a/runtimes/neurun/src/linear/Linear.h b/runtimes/neurun/src/linear/Linear.h
index 16ef785..42a314e 100644 (file)
--- a/runtimes/neurun/src/linear/Linear.h
+++ b/runtimes/neurun/src/linear/Linear.h
@@ -60,6 +60,8 @@ public:
   // TODO Should not return TensorBuilderSet
   backend::TensorBuilderSet planTensors();
 
+  void iterate(const std::function<void(const graph::operation::Node *)> &fn) const;
+
 private:
   const graph::Graph &_graph;
   std::vector<const graph::operation::Node *> _operations;