This commit introduces stride methods and related _stride field in
AvgPool2D class.
Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
#include "coco/IR/Op.h"
#include "coco/IR/Window2D.h"
+#include "coco/IR/Stride2D.h"
namespace coco
{
Window2D *window(void) { return &_window; }
const Window2D *window(void) const { return &_window; }
+public:
+ Stride2D *stride(void) { return &_stride; }
+ const Stride2D *stride(void) const { return &_stride; }
+
private:
void get(const PtrLink<Op, Instr> **out) const override { *out = _op_link; }
private:
Window2D _window;
+ Stride2D _stride;
};
} // namespace coco
// window() SHOULD return a valid pointer
ASSERT_NE(mutable_ptr->window(), nullptr);
ASSERT_EQ(mutable_ptr->window(), immutable_ptr->window());
+
+ // stride() SHOULD return a valid pointer
+ ASSERT_NE(mutable_ptr->stride(), nullptr);
+ ASSERT_EQ(mutable_ptr->stride(), immutable_ptr->stride());
}
TEST_F(AvgPool2DTest, asAvgPool2D)