p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!10575 fix pooling max grad 

From: @wuxuejian
Reviewed-by: @liangchenghui,@oacjiewen
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2020-12-26 16:17:39 +08:00 committed by Gitee
parent 32877aeffb f21e326178
commit 45d6dc716f
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/pooling_max_grad_cpu_kernel.cc
View File

@ -117,15 +117,12 @@ bool MaxPoolingGradCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inpu
size_t src_wh = src_shape_[2] * src_shape_[3];
size_t dst_wh = dst_shape_[2] * dst_shape_[3];
for (size_t n = 0; n < src_shape_[0]; ++n) {
auto task = [&](size_t start, size_t end) {
for (size_t c = start; c < end; ++c) {
ChannelPoolingGrad(input, diff, output);
input = input + src_wh;
output = output + src_wh;
diff = diff + dst_wh;
}
};
CPUKernelUtils::ParallelFor(task, src_shape_[1]);
for (size_t c = 0; c < src_shape_[1]; ++c) {
ChannelPoolingGrad(input, diff, output);
input = input + src_wh;
output = output + src_wh;
diff = diff + dst_wh;
}
}
return true;
}