[onert] Add generating training usedefs for Pool2D op (#13458)

This commit adds generating training usedefs for Pool2D op.

ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>

runtime/onert/core/src/ir/train/UseDefGenerator.cc

@@ -224,6 +224,36 @@ void UseDefGenerator::visit(const train::operation::Pad &node)
   insertBackPropDef(outgoing_index, backwarding_op_index);
 }
 
+void UseDefGenerator::visit(const train::operation::Pool2D &node)
+{
+  if (node.param().op_type != ir::operation::Pool2D::PoolType::MAX)
+  {
+    throw std::runtime_error{"UseDefGenerator: Not yet supported pool type"};
+  }
+
+  assert(_node_to_idx.find(&node) != _node_to_idx.end());
+  const auto &op_index = _node_to_idx.at(&node);
+  const auto backwarding_op_index = TrainingOperationIndex{op_index, false};
+
+  // Insert uses of forwarding output
+  if (node.param().activation != ir::Activation::NONE)
+  {
+    const auto &out_index = node.getOutputs().at(0);
+    const auto out_forwarding_index = TrainingOperandIndex{out_index, true};
+    insertUse(out_forwarding_index, backwarding_op_index);
+  }
+
+  // Insert use of backwarding(backprop) output
+  const auto &out_index = node.getOutputs().at(0);
+  const auto incoming_index = TrainingOperandIndex{out_index, false};
+  insertUse(incoming_index, backwarding_op_index);
+
+  // Set def of backwarding(backprop) input
+  const auto &in_index = node.getInputs().at(train::operation::Pool2D::Input::INPUT);
+  const auto outgoing_index = TrainingOperandIndex{in_index, false};
+  insertBackPropDef(outgoing_index, backwarding_op_index);
+}
+
 void UseDefGenerator::visit(const train::operation::Reshape &node)
 {
   assert(_node_to_idx.find(&node) != _node_to_idx.end());

runtime/onert/core/src/ir/train/UseDefGenerator.h

@@ -68,8 +68,9 @@ class UseDefGenerator : public UseDefGeneratorBase
   void visit(const train::operation::DepthwiseConv2D &node) override;
   void visit(const train::operation::ElementwiseActivation &node) override;
   void visit(const train::operation::Loss &node) override;
-  void visit(const train::operation::Reshape &node) override;
   void visit(const train::operation::Pad &node) override;
+  void visit(const train::operation::Pool2D &node) override;
+  void visit(const train::operation::Reshape &node) override;
 
 private:
   void insertUse(const TrainingOperandIndex &operand_index, const TrainingOperationIndex &op_index);