mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Fix the parameter error of DeformableRoiPool in Ascend device (#2546)

pull/2572/head
liuhw 2023-01-19 21:42:27 +08:00 committed by GitHub
parent dfb03806a1
commit 0b005c52b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp
View File

@ -45,11 +45,11 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois,
int64_t sampling_ratio_ = (int64_t)sampling_ratio;
OpCommand cmd;
cmd.Name("DeformableRoiPoolGrad")
.Input(grad_input)
.Input(grad_output)
.Input(input)
.Input(rois)
.Input(offset)
.Output(grad_output)
.Output(grad_input)
.Output(grad_offset)
.Attr("output_size", output_size)
.Attr("spatial_scale", spatial_scale)