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
2 merge requests!105version 0.2.0,!77Support for recurrent networks
Pipeline #40270 passed
...@@ -74,8 +74,17 @@ public: ...@@ -74,8 +74,17 @@ public:
} }
void computeOutputDims() override final { void computeOutputDims() override final {
// Forward dims of micro-graph // Check first that all required inputs are available, otherwise
mGraph->forwardDims(); // 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