From 7df87c11b0f2548ece5f086e1f4d571ed85d5fe0 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Mon, 22 Apr 2024 15:35:00 +0200
Subject: [PATCH] fix computeOutputDims
---
src/operator/Reshape.cpp | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/operator/Reshape.cpp b/src/operator/Reshape.cpp
index ab53c094d..624524b3b 100644
--- a/src/operator/Reshape.cpp
+++ b/src/operator/Reshape.cpp
@@ -54,6 +54,10 @@ bool Aidge::Reshape_Op::forwardDims(bool /*allowDataDependency*/) {
dimSize = 1;
negativeIndex = static_cast<DimIdx_t>(i);
}
+ else if (dimSize == 0)
+ {
+ dimSize = getInput(0) -> dims()[i];
+ }
outDims.push_back(static_cast<DimSize_t>(dimSize));
outSize *= static_cast<DimSize_t>(dimSize);
}
--
GitLab