Skip to content
Snippets Groups Projects
Commit 4ae7f3aa authored by Noam Zerah's avatar Noam Zerah Committed by Olivier BICHLER
Browse files

Fix avgpooling forwardims when kernel size > input dim size

parent 2f2535bb
No related branches found
No related tags found
3 merge requests!414Update version 0.5.1 -> 0.6.0,!408[Add] Dropout Operator,!382Fix avgpooling forwardims when kernel size > input dim size
Pipeline #69516 passed
......@@ -75,11 +75,12 @@ bool Aidge::AvgPooling_Op<DIM>::forwardDims(bool /*allowDataDependency*/) {
const auto strideDim = mAttributes->template getAttr<AvgPoolingAttr::StrideDims>()[dim];
const auto dilationDim = mAttributes->template getAttr<AvgPoolingAttr::Dilations>()[dim];
outputDims[dim+2] = 1 + static_cast<DimSize_t>(
roundingFunction(static_cast<float>(inputDims[dim+2] -
(kernelDim - 1) * dilationDim - 1) /
static_cast<float>(strideDim)));
const float effective_size = static_cast<float>(inputDims[dim+2] - (kernelDim - 1) * dilationDim - 1);
const float rounded_val = roundingFunction(effective_size / static_cast<float>(strideDim));
outputDims[dim+2] = 1 + std::max(0, static_cast<int>(rounded_val));
}
outputDims[1] = inputDims[1];
outputDims[0] = inputDims[0];
getOutput(0)->resize(outputDims);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment