[enco] Build IR from Max pooling layer with padding (#1378)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Thu, 6 Sep 2018 05:37:46 +0000 (14:37 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 6 Sep 2018 05:37:46 +0000 (14:37 +0900)
This commit revises enco caffe frontend to update padding on MaxPool2D
op with the padding values from pooling layer specification.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
contrib/enco/frontend/caffe/src/Frontend.cpp

index 6ecfbce..1d46cfe 100644 (file)
@@ -310,6 +310,11 @@ enco::Bundle Frontend::load(void) const
         op->stride()->vertical(spec.vertical_stride());
         op->stride()->horizontal(spec.horizontal_stride());
 
+        op->pad()->top(spec.vertical_pad());
+        op->pad()->bottom(spec.vertical_pad());
+        op->pad()->left(spec.horizontal_pad());
+        op->pad()->right(spec.horizontal_pad());
+
         return op;
       };