[enco.caffe] Set MaxPool2D op stride values (#1323)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Tue, 4 Sep 2018 08:22:06 +0000 (17:22 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Tue, 4 Sep 2018 08:22:06 +0000 (17:22 +0900)
With this commit, Caffe frontend now correctly sets stride of
MaxPool2D operation using the values inferred from caffe model.

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

index 8a574d8..249ac26 100644 (file)
@@ -268,6 +268,9 @@ enco::Bundle Frontend::load(void) const
       op->window()->vertical(spec.window_height());
       op->window()->horizontal(spec.window_width());
 
+      op->stride()->vertical(spec.vertical_stride());
+      op->stride()->horizontal(spec.horizontal_stride());
+
       // Create a UnitF instruction
       auto ins = m->entity()->instr()->create<coco::UnitF>();