Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 e8d171bb 编写于 作者: X xzl
浏览文件
操作

add check for groups and inputChannels

上级 cfd4c05e
显示空白变更内容
内联 并排
Showing with 3 addition and 0 deletion
paddle/function/DepthwiseConvOp.cpp
浏览文件 @ e8d171bb
...@@ -125,6 +125,7 @@ public: ...@@ -125,6 +125,7 @@ public:
size_t outputHeight = output[2]; size_t outputHeight = output[2];
size_t outputWidth = output[3]; size_t outputWidth = output[3];
size_t filterMultiplier = outputChannels / groups_; size_t filterMultiplier = outputChannels / groups_;
CHECK_EQ(inputChannels, groups_);
real* inputData = inputs[0].data<real>(); real* inputData = inputs[0].data<real>();
real* filterData = inputs[1].data<real>(); real* filterData = inputs[1].data<real>();
...@@ -187,6 +188,7 @@ public: ...@@ -187,6 +188,7 @@ public:
size_t outputHeight = output[2]; size_t outputHeight = output[2];
size_t outputWidth = output[3]; size_t outputWidth = output[3];
size_t filterMultiplier = outputChannels / groups_; size_t filterMultiplier = outputChannels / groups_;
CHECK_EQ(inputChannels, groups_);
real* outputGrad = inputs[0].data<real>(); real* outputGrad = inputs[0].data<real>();
real* filterData = inputs[1].data<real>(); real* filterData = inputs[1].data<real>();
...@@ -248,6 +250,7 @@ public: ...@@ -248,6 +250,7 @@ public:
size_t outputHeight = output[2]; size_t outputHeight = output[2];
size_t outputWidth = output[3]; size_t outputWidth = output[3];
size_t filterMultiplier = outputChannels / groups_; size_t filterMultiplier = outputChannels / groups_;
CHECK_EQ(inputChannels, groups_);
real* outputGrad = inputs[0].data<real>(); real* outputGrad = inputs[0].data<real>();
real* inputData = inputs[1].data<real>(); real* inputData = inputs[1].data<real>();
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或