From 73cc1b3b36899a7cb24480837cc06669774fce82 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Tue, 14 Nov 2023 00:47:01 +0000
Subject: [PATCH] [Fix] Attributes first index result in Conv
---
include/aidge/operator/Conv.hpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/operator/Conv.hpp b/include/aidge/operator/Conv.hpp
index a9f62a5fd..1d4edf225 100644
--- a/include/aidge/operator/Conv.hpp
+++ b/include/aidge/operator/Conv.hpp
@@ -171,8 +171,8 @@ public:
// Result
std::vector<std::pair<std::size_t, std::vector<DimSize_t>>> res;
res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[0]->getIdx(inputIdxDims), inputDims));
- res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(weightIdxDims, weightDims));
- res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(biasIdxDims, biasDims));
+ res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[1]->getIdx(weightIdxDims), weightDims));
+ res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[2]->getIdx(biasIdxDims), biasDims));
return res;
}
AIDGE_THROW_OR_ABORT(std::runtime_error, "Given outputDim out of range or output dim not forwarded yet.");
--
GitLab