Skip to content
Snippets Groups Projects
Commit b7a41332 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fix attrs

parent e9e0122f
No related branches found
No related tags found
2 merge requests!212Version 0.3.0,!153Im2col
Pipeline #50110 failed
...@@ -73,12 +73,12 @@ bool Aidge::Reshape_Op::forwardDims(bool allowDataDependency) { ...@@ -73,12 +73,12 @@ bool Aidge::Reshape_Op::forwardDims(bool allowDataDependency) {
{ {
int64_t dimSize = this->shape()[i]; int64_t dimSize = this->shape()[i];
if (dimSize < 0) { if (dimSize < 0) {
AIDGE_ASSERT(!foundNegativeDimension, "Found more than one negative dimension in Reshape Operator: {}.", this->template getAttr<ReshapeAttr::Shape>()); AIDGE_ASSERT(!foundNegativeDimension, "Found more than one negative dimension in Reshape Operator: {}.", this->shape());
foundNegativeDimension = true; foundNegativeDimension = true;
dimSize = 1; dimSize = 1;
negativeIndex = static_cast<DimIdx_t>(i); negativeIndex = static_cast<DimIdx_t>(i);
} }
else if (dimSize == 0 && !mAttributes->template getAttr<ReshapeAttr::AllowZero>()) else if (dimSize == 0 && !this->allowZero())
{ {
dimSize = getInput(0) -> dims()[i]; dimSize = getInput(0) -> dims()[i];
} }
......
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