Skip to content

GitLab

Unverified Commit 0b005c52 authored by liuhw's avatar liuhw Committed by GitHub
Browse files

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

parent dfb03806
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/ops/csrc/pytorch/npu/deform_roi_pool.cpp
View file @ 0b005c52
...@@ -45,11 +45,11 @@ void deform_roi_pool_backward_npu(Tensor grad_output, Tensor input, Tensor rois, ...@@ -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; int64_t sampling_ratio_ = (int64_t)sampling_ratio;
OpCommand cmd; OpCommand cmd;
cmd.Name("DeformableRoiPoolGrad") cmd.Name("DeformableRoiPoolGrad")
.Input(grad_input) .Input(grad_output)
.Input(input) .Input(input)
.Input(rois) .Input(rois)
.Input(offset) .Input(offset)
.Output(grad_output) .Output(grad_input)
.Output(grad_offset) .Output(grad_offset)
.Attr("output_size", output_size) .Attr("output_size", output_size)
.Attr("spatial_scale", spatial_scale) .Attr("spatial_scale", spatial_scale)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment