From 03ac8a73c317787335c2aae3c54075498124224f 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: Thu, 6 Sep 2018 13:38:36 +0900 Subject: [PATCH] [enco] Build IR from Average Pooling layers with padding (#1373) This commit extends enco caffe frontend to support coco IR build on average pooling layers with padding. Signed-off-by: Jonghyun Park --- contrib/enco/frontend/caffe/src/Frontend.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/contrib/enco/frontend/caffe/src/Frontend.cpp b/contrib/enco/frontend/caffe/src/Frontend.cpp index 50182f2..6ecfbce 100644 --- a/contrib/enco/frontend/caffe/src/Frontend.cpp +++ b/contrib/enco/frontend/caffe/src/Frontend.cpp @@ -317,12 +317,20 @@ enco::Bundle Frontend::load(void) const builders[PoolingMethod::Avg] = [](coco::Module *m, const PoolingSpec &spec) { auto op = m->entity()->op()->create(); + // NOTE Caffe use static divisor on average pooling + op->divisor(coco::AvgPool2D::Divisor::Static); + op->window()->vertical(spec.window_height()); op->window()->horizontal(spec.window_height()); 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; }; -- 2.7.4