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

[Fix] Fix deform_conv inconsistent pybind (#1781)

pull/1802/head
Jiazhen Wang 2022-03-15 11:15:15 +08:00 committed by GitHub
parent 2eb0a10d5a
commit bdabdffcf9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmcv/ops/csrc/pytorch/pybind.cpp
View File

@ -470,21 +470,21 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("deform_conv_forward", &deform_conv_forward, "deform_conv_forward",
py::arg("input"), py::arg("weight"), py::arg("offset"),
py::arg("output"), py::arg("columns"), py::arg("ones"), py::arg("kW"),
py::arg("kH"), py::arg("dW"), py::arg("dH"), py::arg("padH"),
py::arg("padW"), py::arg("dilationW"), py::arg("dilationH"),
py::arg("kH"), py::arg("dW"), py::arg("dH"), py::arg("padW"),
py::arg("padH"), py::arg("dilationW"), py::arg("dilationH"),
py::arg("group"), py::arg("deformable_group"), py::arg("im2col_step"));
m.def("deform_conv_backward_input", &deform_conv_backward_input,
"deform_conv_backward_input", py::arg("input"), py::arg("offset"),
py::arg("gradOutput"), py::arg("gradInput"), py::arg("gradOffset"),
py::arg("weight"), py::arg("columns"), py::arg("kW"), py::arg("kH"),
py::arg("dW"), py::arg("dH"), py::arg("padH"), py::arg("padW"),
py::arg("dW"), py::arg("dH"), py::arg("padW"), py::arg("padH"),
py::arg("dilationW"), py::arg("dilationH"), py::arg("group"),
py::arg("deformable_group"), py::arg("im2col_step"));
m.def("deform_conv_backward_parameters", &deform_conv_backward_parameters,
"deform_conv_backward_parameters", py::arg("input"), py::arg("offset"),
py::arg("gradOutput"), py::arg("gradWeight"), py::arg("columns"),
py::arg("ones"), py::arg("kW"), py::arg("kH"), py::arg("dW"),
py::arg("dH"), py::arg("padH"), py::arg("padW"), py::arg("dilationW"),
py::arg("dH"), py::arg("padW"), py::arg("padH"), py::arg("dilationW"),
py::arg("dilationH"), py::arg("group"), py::arg("deformable_group"),
py::arg("scale"), py::arg("im2col_step"));
m.def("deform_roi_pool_forward", &deform_roi_pool_forward,