Skip to content
Snippets Groups Projects
Commit 62de2285 authored by Maxence Naud's avatar Maxence Naud
Browse files

Merge remote-tracking branch 'origin/fowarddims' into fowarddims

parents 68549dad 7a944253
No related branches found
No related tags found
No related merge requests found
......@@ -54,7 +54,9 @@ bool Aidge::AvgPooling_Op<DIM>::forwardDims(bool /*allowDataDependency*/) {
static_cast<float>(this->template getAttr<AvgPoolingAttr::StrideDims>()[dim])));
}
getOutput(0)->resize(outputDims);
return true;
}
return false;
}
......
......@@ -53,6 +53,7 @@ bool Aidge::BatchNorm_Op<DIM>::forwardDims(bool /*allowDataDependency*/) {
}
mOutputs[0]->resize(getInput(0)->dims());
}
return associated;
}
template <Aidge::DimIdx_t DIM>
......
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