[neurun] Introduce more operation methods (#2777)

Introduce methods for replacing operands for `operation::Node`.

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

diff --git a/runtimes/neurun/src/graph/operation/Node.cc b/runtimes/neurun/src/graph/operation/Node.cc
index f472bc0..eeb3ff4 100644 (file)
--- a/runtimes/neurun/src/graph/operation/Node.cc
+++ b/runtimes/neurun/src/graph/operation/Node.cc
@@ -29,6 +29,16 @@ Node::Node() = default;
 
 Node::~Node() = default;
 
+void Node::replaceInput(const operand::Index &from, const operand::Index &to)
+{
+  _inputs.replace(from, to);
+}
+
+void Node::replaceOutput(const operand::Index &from, const operand::Index &to)
+{
+  _outputs.replace(from, to);
+}
+
 void Node::lower_info(std::unique_ptr<LowerInfo> &&lower_info)
 {
   _lower_info = std::move(lower_info);

diff --git a/runtimes/neurun/src/graph/operation/Node.h b/runtimes/neurun/src/graph/operation/Node.h
index 242e05a..bab51b3 100644 (file)
--- a/runtimes/neurun/src/graph/operation/Node.h
+++ b/runtimes/neurun/src/graph/operation/Node.h
@@ -51,6 +51,8 @@ public:
   virtual std::string getName() const = 0;
 
 public:
+  virtual void replaceInput(const operand::Index &from, const operand::Index &to);
+  virtual void replaceOutput(const operand::Index &from, const operand::Index &to);
   virtual const operand::IndexSet &getInputs() const { return _inputs; }
   virtual const operand::IndexSet &getOutputs() const { return _outputs; }
   // It's for only input/output tensors but const data.