explicit PoolOp(const Shape &windowShape, const Shape &strides, PoolingType poolType,
PaddingType padding)
- : OpDescription(1, 1), _padding(padding), _poolingType(poolType), _strides(strides),
- _windowShape(windowShape)
+ : OpDescription(1, 1), _padding(padding), _poolingType(poolType), _windowShape(windowShape), _strides(strides)
{
_pads.resize(_windowShape.rank());
}
const Shape &getStrides() const { return _strides; }
- const int getPadding(int dim) const { return _pads[dim]; }
+ int getPadding(int dim) const { return _pads[dim]; }
void setPadding(int dim, int pad) { _pads[dim] = pad; }