fix SwitchOrderLayer forward

8 years ago · e76eb8534d
parent bc9f4c5590
commit e76eb8534d
1 changed files with 1 additions and 2 deletions
--- a/paddle/gserver/layers/SwitchOrderLayer.cpp
+++ b/paddle/gserver/layers/SwitchOrderLayer.cpp
@ -83,8 +83,7 @@ void SwitchOrderLayer::forward(PassType passType) {
  setOutDims();
  resetOutput(outDims_[0], outDims_[1] * outDims_[2] * outDims_[3]);
  if (heightAxis_.size() > 0) {
-    getOutputValue()->reshape(reshapeHeight_, reshapeWidth_);
-    getOutputGrad()->reshape(reshapeHeight_, reshapeWidth_);
+    resetOutput(reshapeHeight_, reshapeWidth_);
  }

  // switch NCHW to NHWC