[coco] Create MaxPool2D with OpManager (#1310)

This commit implements 'create' method for 'MaxPool2D' op in
'OpManager'.

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

diff --git a/contrib/coco/core/include/coco/IR/OpManager.h b/contrib/coco/core/include/coco/IR/OpManager.h
index 83b95fc..e442b32 100644 (file)
--- 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 (file)
--- a/contrib/coco/core/src/IR/OpManager.cpp
+++ b/contrib/coco/core/src/IR/OpManager.cpp
@@ -18,6 +18,11 @@ template <> Conv2D *OpManager::create<Conv2D>(void)
   return take(make_unique<Conv2D>(_op_link, _obj_link));
 }
 
+template <> MaxPool2D *OpManager::create<MaxPool2D>(void)
+{
+  return take(make_unique<MaxPool2D>(_op_link));
+}
+
 template <> ReLU *OpManager::create<ReLU>(void) { return take(make_unique<ReLU>(_op_link)); }
 
 } // namespace coco