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

!4481 ROIAlignGrad kernel output size is wrong 

Merge pull request !4481 from JonathanY/roiwrapper
This commit is contained in:
mindspore-ci-bot 2020-08-15 14:00:33 +08:00 committed by Gitee
parent 9f635e52c7 19d7488195
commit 2e79ff2eef
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/roi_align_grad_gpu_kernel.h
View File

@ -91,13 +91,14 @@ class ROIAlignGradGpuFwdKernel : public GpuKernel {
roi_end_mode_ = 1;
// Get channels, height & width
batch_size_ = xdiff_shape_[0];
channels_ = xdiff_shape_[1];
height_ = xdiff_shape_[2];
width_ = xdiff_shape_[3];
// Get output_shape
output_shape_ = {roi_rows_, channels_, height_, width_};
output_size_ = roi_rows_ * channels_ * height_ * width_ * sizeof(T);
output_shape_ = {batch_size_, channels_, height_, width_};
output_size_ = batch_size_ * channels_ * height_ * width_ * sizeof(T);
InitSizeLists();
return true;
@ -120,6 +121,7 @@ class ROIAlignGradGpuFwdKernel : public GpuKernel {
int roi_rows_;
int roi_cols_;
int batch_size_;
int channels_;
int height_;
int width_;