[coco] Introduce padding-related members to MaxPool2D (#1376)

This commit introduces padding-related members to MaxPool2D class to
allow frontend to deliver padding information to backend.

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

diff --git a/contrib/coco/core/include/coco/IR/MaxPool2D.h b/contrib/coco/core/include/coco/IR/MaxPool2D.h
index 4b1288d..b555158 100644 (file)
--- a/contrib/coco/core/include/coco/IR/MaxPool2D.h
+++ b/contrib/coco/core/include/coco/IR/MaxPool2D.h
@@ -4,6 +4,7 @@
 #include "coco/IR/Op.h"
 #include "coco/IR/Window2D.h"
 #include "coco/IR/Stride2D.h"
+#include "coco/IR/Padding2D.h"
 
 namespace coco
 {
@@ -38,6 +39,10 @@ public:
   Stride2D *stride(void) { return &_stride; }
   const Stride2D *stride(void) const { return &_stride; }
 
+public:
+  Padding2D *pad(void) { return &_pad; }
+  const Padding2D *pad(void) const { return &_pad; }
+
 private:
   void get(const PtrLink<Op, Instr> **out) const override { *out = _op_link; }
 
@@ -47,6 +52,7 @@ private:
 private:
   Window2D _window;
   Stride2D _stride;
+  Padding2D _pad;
 };
 
 } // namespace coco

diff --git a/contrib/coco/core/src/IR/MaxPool2D.test.cpp b/contrib/coco/core/src/IR/MaxPool2D.test.cpp
index 2715ebc..8ad1525 100644 (file)
--- a/contrib/coco/core/src/IR/MaxPool2D.test.cpp
+++ b/contrib/coco/core/src/IR/MaxPool2D.test.cpp
@@ -55,6 +55,10 @@ TEST_F(MaxPool2DTest, initialization)
   // stride() SHOULD return a valid pointer
   ASSERT_NE(mutable_ptr->stride(), nullptr);
   ASSERT_EQ(mutable_ptr->stride(), immutable_ptr->stride());
+
+  // pad() SHOULD return a valid pointer
+  ASSERT_NE(mutable_ptr->pad(), nullptr);
+  ASSERT_EQ(mutable_ptr->pad(), immutable_ptr->pad());
 }
 
 TEST_F(MaxPool2DTest, asMaxPool2D)