Skip to content
Snippets Groups Projects
Commit dba5e80d authored by Jerome Hue's avatar Jerome Hue
Browse files

Fix StackOp forward()

Call OperatorTensor::forward() instead of Operator::forward(), ensuring dims are
forwarded.
parent 03de25a0
No related branches found
No related tags found
3 merge requests!414Update version 0.5.1 -> 0.6.0,!408[Add] Dropout Operator,!342feat: Backward of Stack Operator
...@@ -132,7 +132,7 @@ std::set<std::string> StackOp::getAvailableBackends() const { ...@@ -132,7 +132,7 @@ std::set<std::string> StackOp::getAvailableBackends() const {
} }
void StackOp::forward() { void StackOp::forward() {
Operator::forward(); OperatorTensor::forward();
++forwardStep(); ++forwardStep();
} }
......
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