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

Fixed a bug with meta op computeOutputDims()

parent 7161e363
No related branches found
No related tags found
No related merge requests found
......@@ -74,8 +74,17 @@ public:
}
void computeOutputDims() override final {
// Forward dims of micro-graph
mGraph->forwardDims();
// Check first that all required inputs are available, otherwise
// mGraph->forwardDims() will fail!
bool forwarded = true;
for (IOIndex_t i = 0; i < nbInputs(); ++i) {
forwarded &= mInputs[i] ? !(getInput(i)->empty()) : false;
}
if (forwarded) {
// Forward dims of micro-graph
mGraph->forwardDims();
}
}
......
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