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*/) { ...@@ -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>(); const std::array<DimSize_t, DIM + 2> inputDims = getInput(0)->template dims<DIM+2>();
for (std::size_t dim = 0; dim < DIM; ++dim) { 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] + inputDims[dim+2]
+ mAttributes->template getAttr<PadAttr::BeginEndBorders>()[2*dim+1]; + mAttributes->template getAttr<PadAttr::BeginEndBorders>()[DIM+dim];
} }
outputDims[1] = inputDims[1]; outputDims[1] = inputDims[1];
outputDims[0] = inputDims[0]; 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