[onert] Introduce replacing an input

runtime/onert/core/include/ir/IOperation.h

@@ -38,6 +38,7 @@ struct IOperation
   virtual std::string name() const { return std::string{toString(opcode())}; }
   virtual OpCode opcode() const = 0;
 
+  virtual void replaceInput(size_t pos, const OperandIndex &index) = 0;
   virtual void replaceInputs(const OperandIndex &from, const OperandIndex &to) = 0;
   virtual void replaceOutputs(const OperandIndex &from, const OperandIndex &to) = 0;
   virtual const OperandIndexSequence &getInputs() const = 0;

runtime/onert/core/include/ir/OperandIndexSequence.h

@@ -50,6 +50,7 @@ class OperandIndexSequence
   const OperandIndex &at(IOIndex set_index) const { return _vec.at(set_index.value()); }
   const OperandIndex &at(uint32_t index) const { return _vec.at(index); }
   bool contains(const OperandIndex &index) const;
+  void replace(size_t pos, const OperandIndex &index);
   void replace(const OperandIndex &from, const OperandIndex &to);
   OperandIndexSequence operator|(ir::Remove filter) const
   {

runtime/onert/core/include/ir/Operation.h

@@ -48,6 +48,7 @@ class Operation : virtual public IOperation
   virtual ~Operation();
 
 public:
+  void replaceInput(size_t pos, const OperandIndex &index) override;
   void replaceInputs(const OperandIndex &from, const OperandIndex &to) override;
   void replaceOutputs(const OperandIndex &from, const OperandIndex &to) override;
   OperandIndexSequence &getInputs() { return _inputs; }

runtime/onert/core/src/ir/OperandIndexSequence.cc

@@ -17,6 +17,7 @@
 #include "ir/OperandIndexSequence.h"
 
 #include <algorithm>
+#include <cassert>
 #include <sstream>
 
 namespace onert
@@ -50,6 +51,12 @@ bool OperandIndexSequence::contains(const OperandIndex &index) const
   return std::find(_vec.begin(), _vec.end(), index) != _vec.end();
 }
 
+void OperandIndexSequence::replace(size_t pos, const OperandIndex &index)
+{
+  assert(pos < _vec.size() && "OperandIndexSequence: Out of range");
+  _vec.at(pos) = index;
+}
+
 void OperandIndexSequence::replace(const OperandIndex &from, const OperandIndex &to)
 {
   std::replace(_vec.begin(), _vec.end(), from, to);

runtime/onert/core/src/ir/Operation.cc

@@ -52,6 +52,8 @@ void Operation::setOutputs(const OperandIndexSequence &indexes)
   _outputs = indexes;
 }
 
+void Operation::replaceInput(size_t pos, const OperandIndex &index) { _inputs.replace(pos, index); }
+
 void Operation::replaceInputs(const OperandIndex &from, const OperandIndex &to)
 {
   _inputs.replace(from, to);