未验证 提交 020630b7 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #7395 from tensor-tang/fixds2

fix ds2 issue
...@@ -132,6 +132,8 @@ void MKLDNNLayer::reshapeInput(int& batchsize, ...@@ -132,6 +132,8 @@ void MKLDNNLayer::reshapeInput(int& batchsize,
if (w != 0) { if (w != 0) {
width = w; width = w;
} }
height = height != 0 ? height : 1;
width = width != 0 ? width : 1;
} }
void MKLDNNLayer::reshapeOutput(size_t height, size_t width) { void MKLDNNLayer::reshapeOutput(size_t height, size_t width) {
......
...@@ -98,6 +98,8 @@ protected: ...@@ -98,6 +98,8 @@ protected:
public: public:
explicit MKLDNNLayer(const LayerConfig& config) explicit MKLDNNLayer(const LayerConfig& config)
: Layer(config), : Layer(config),
ih_(0),
iw_(0),
condition_(0), condition_(0),
needResetBwd_(true), needResetBwd_(true),
outputOnlyMKLDNN_(false), outputOnlyMKLDNN_(false),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册