From 01ba78ffd485b2ebc201264b4574b41e098a99d1 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Tue, 4 Sep 2018 10:25:53 +0900 Subject: [PATCH] [coco] Create MaxPool2D with OpManager (#1310) This commit implements 'create' method for 'MaxPool2D' op in 'OpManager'. Signed-off-by: Jonghyun Park --- contrib/coco/core/include/coco/IR/OpManager.h | 1 + contrib/coco/core/src/IR/OpManager.cpp | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/contrib/coco/core/include/coco/IR/OpManager.h b/contrib/coco/core/include/coco/IR/OpManager.h index 83b95fc..e442b32 100644 --- a/contrib/coco/core/include/coco/IR/OpManager.h +++ b/contrib/coco/core/include/coco/IR/OpManager.h @@ -3,6 +3,7 @@ #include "coco/IR/Op.h" #include "coco/IR/Conv2D.h" +#include "coco/IR/MaxPool2D.h" #include "coco/IR/ReLU.h" #include "coco/IR/Instr.forward.h" diff --git a/contrib/coco/core/src/IR/OpManager.cpp b/contrib/coco/core/src/IR/OpManager.cpp index 3786b64..77c8543 100644 --- a/contrib/coco/core/src/IR/OpManager.cpp +++ b/contrib/coco/core/src/IR/OpManager.cpp @@ -18,6 +18,11 @@ template <> Conv2D *OpManager::create(void) return take(make_unique(_op_link, _obj_link)); } +template <> MaxPool2D *OpManager::create(void) +{ + return take(make_unique(_op_link)); +} + template <> ReLU *OpManager::create(void) { return take(make_unique(_op_link)); } } // namespace coco -- 2.7.4