From 8dd047c2b39450127417fa56feae9f2e288c1ccd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Thu, 30 Jan 2025 15:19:24 +0100
Subject: [PATCH] chore : [Conv] updated the method for attribute accession to
ease code reading
---
src/operator/Conv.cpp | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/operator/Conv.cpp b/src/operator/Conv.cpp
index ee850414c..be575a1f2 100644
--- a/src/operator/Conv.cpp
+++ b/src/operator/Conv.cpp
@@ -105,18 +105,18 @@ Aidge::Conv_Op<DIM>::computeReceptiveField(
std::vector<DimSize_t> inputDims{outputDims[0], getInput(0)->dims()[1]};
for (DimIdx_t i = 0; i < DIM; ++i) {
inputDims.push_back((outputDims[2+static_cast<std::size_t>(i)] - 1)
- * mAttributes->template getAttr<ConvAttr::StrideDims>()[static_cast<std::size_t>(i)]
+ * strideDims()[static_cast<std::size_t>(i)]
+ 1
- + (mAttributes->template getAttr<ConvAttr::KernelDims>()[static_cast<std::size_t>(i)] - 1)
- * mAttributes->template getAttr<ConvAttr::DilationDims>()[static_cast<std::size_t>(i)]);
- inputIdxDims[2+i] *= mAttributes->template getAttr<ConvAttr::StrideDims>()[static_cast<std::size_t>(i)];
+ + (kernelDims()[static_cast<std::size_t>(i)] - 1)
+ * dilationDims()[static_cast<std::size_t>(i)]);
+ inputIdxDims[2+i] *= strideDims()[static_cast<std::size_t>(i)];
}
// Weight
// same output value, every input channel is used
std::vector<DimSize_t> weightDims{outputDims[1], getInput(0)->dims()[1]};
for (std::size_t i = 0; i < DIM; ++i) {
- weightDims.push_back(mAttributes->template getAttr<ConvAttr::KernelDims>()[i]);
+ weightDims.push_back(kernelDims()[i]);
}
std::vector<DimSize_t> weightIdxDims = std::vector<DimSize_t>(DIM+2, 0);
weightIdxDims[0] = firstEltDims[1];
--
GitLab