From 901992117bfb0a0cd9f6978f240442aa4b237383 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: Wed, 5 Sep 2018 18:06:26 +0900 Subject: [PATCH] [enco] Set AvgPool2D stride on ANN subnet construction (#1349) With this commit, ANN subnet builder uses stride values stored in coco IR to set strides of ANN AvgPool2D operation. Signed-off-by: Jonghyun Park --- contrib/enco/core/src/Transforms/Split.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/contrib/enco/core/src/Transforms/Split.cpp b/contrib/enco/core/src/Transforms/Split.cpp index 1e866fe..ea08241 100644 --- a/contrib/enco/core/src/Transforms/Split.cpp +++ b/contrib/enco/core/src/Transforms/Split.cpp @@ -301,11 +301,11 @@ public: auto bottom = _binder->addOperand(); _binder->setOperand(bottom, 0); - // TODO Support horizontal/vertical stride + // Set horizontal/vertical stride auto hstride = _binder->addOperand(); - _binder->setOperand(hstride, 1); + _binder->setOperand(hstride, avgpool->stride()->horizontal()); auto vstride = _binder->addOperand(); - _binder->setOperand(vstride, 1); + _binder->setOperand(vstride, avgpool->stride()->vertical()); // Set receptive field size auto width = _binder->addOperand(); -- 2.7.4