Skip to content
Snippets Groups Projects
Commit ebca4402 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fix padding

parent 98e64c9b
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!232Fix padding
Pipeline #57619 passed
......@@ -34,9 +34,9 @@ bool Aidge::Pad_Op<DIM>::forwardDims(bool /*allowDataDependency*/) {
const std::array<DimSize_t, DIM + 2> inputDims = getInput(0)->template dims<DIM+2>();
for (std::size_t dim = 0; dim < DIM; ++dim) {
outputDims[dim+2] = mAttributes->template getAttr<PadAttr::BeginEndBorders>()[2*dim]
outputDims[dim+2] = mAttributes->template getAttr<PadAttr::BeginEndBorders>()[dim]
+ inputDims[dim+2]
+ mAttributes->template getAttr<PadAttr::BeginEndBorders>()[2*dim+1];
+ mAttributes->template getAttr<PadAttr::BeginEndBorders>()[DIM+dim];
}
outputDims[1] = inputDims[1];
outputDims[0] = inputDims[0];
......
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