From ebca440240effd30eedcc4703ad837fe5be8a665 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 23 Oct 2024 15:15:36 +0200
Subject: [PATCH] Fix padding
---
src/operator/Pad.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/Pad.cpp b/src/operator/Pad.cpp
index ba762da57..a0b5f2df5 100644
--- a/src/operator/Pad.cpp
+++ b/src/operator/Pad.cpp
@@ -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];
--
GitLab