builders[PoolingMethod::Max] = [](coco::Module *m, const PoolingSpec &spec) {
auto op = m->entity()->op()->create<coco::MaxPool2D>();
- op->window()->vertical(spec.window_height());
- op->window()->horizontal(spec.window_width());
+ op->window()->height(spec.window_height());
+ op->window()->width(spec.window_width());
op->stride()->vertical(spec.vertical_stride());
op->stride()->horizontal(spec.horizontal_stride());
// 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->window()->height(spec.window_height());
+ // TODO FIX THIS BUG AND ADD TEST
+ op->window()->width(spec.window_height());
op->stride()->vertical(spec.vertical_stride());
op->stride()->horizontal(spec.horizontal_stride());