fix ds2 issue

7 years ago · a7e847b648
parent df92776850
commit a7e847b648
2 changed files with 4 additions and 0 deletions
--- a/paddle/gserver/layers/MKLDNNLayer.cpp
+++ b/paddle/gserver/layers/MKLDNNLayer.cpp
@ -132,6 +132,8 @@ void MKLDNNLayer::reshapeInput(int& batchsize,
  if (w != 0) {
    width = w;
  }
+  height = height != 0 ? height : 1;
+  width = width != 0 ? width : 1;
 }

 void MKLDNNLayer::reshapeOutput(size_t height, size_t width) {
--- a/paddle/gserver/layers/MKLDNNLayer.h
+++ b/paddle/gserver/layers/MKLDNNLayer.h
@ -98,6 +98,8 @@ protected:
 public:
  explicit MKLDNNLayer(const LayerConfig& config)
      : Layer(config),
+        ih_(0),
+        iw_(0),
        condition_(0),
        needResetBwd_(true),
        outputOnlyMKLDNN_(false),