[enco] Configure ANN subnet MaxPool2D strides (#1321)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Tue, 4 Sep 2018 07:41:31 +0000 (16:41 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Tue, 4 Sep 2018 07:41:31 +0000 (16:41 +0900)
With this commit, enco now sets stride values for MaxPool2D op in android
NN subnet from stride values stored in coco::MaxPool2D entity.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
contrib/enco/core/src/Transforms/Split.cpp

index 46d6317..94361ff 100644 (file)
@@ -258,11 +258,11 @@ public:
       auto bottom = _binder->addOperand<int32_t>();
       _binder->setOperand(bottom, 0);
 
-      // TODO Support stride
+      // Set horizontal/vertical stride
       auto hstride = _binder->addOperand<int32_t>();
-      _binder->setOperand(hstride, 1);
+      _binder->setOperand(hstride, maxpool->stride()->horizontal());
       auto vstride = _binder->addOperand<int32_t>();
-      _binder->setOperand(vstride, 1);
+      _binder->setOperand(vstride, maxpool->stride()->vertical());
 
       // Set receptive field size
       auto width = _binder->addOperand<int32_t>();