Merge pull request #13844 from jerrywgz/fix_roi_pool

fix roi pool register

Merge pull request #13844 from jerrywgz/fix_roi_pool
fix roi pool register
a1d3db03 · jerrywgz · GitHub · 0a9f5f17 · 55334262 · a1d3db03
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/operators/roi_pool_op.cc paddle/fluid/operators/roi_pool_op.cc +1 -1

paddle/fluid/operators/roi_pool_op.cu paddle/fluid/operators/roi_pool_op.cu +1 -1

未找到文件。
--- a/paddle/fluid/operators/roi_pool_op.cc
+++ b/paddle/fluid/operators/roi_pool_op.cc
@@ -174,4 +174,4 @@ REGISTER_OP_CPU_KERNEL(
 REGISTER_OP_CPU_KERNEL(
    roi_pool_grad,
    ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, float>,
-    ops::CPUROIPoolOpKernel<paddle::platform::CPUDeviceContext, double>);
+    ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>);
--- a/paddle/fluid/operators/roi_pool_op.cu
+++ b/paddle/fluid/operators/roi_pool_op.cu
@@ -249,4 +249,4 @@ REGISTER_OP_CUDA_KERNEL(
 REGISTER_OP_CUDA_KERNEL(
    roi_pool_grad,
    ops::GPUROIPoolGradOpKernel<paddle::platform::CUDADeviceContext, float>,
-    ops::GPUROIPoolOpKernel<paddle::platform::CUDADeviceContext, double>);
+    ops::GPUROIPoolGradOpKernel<paddle::platform::CUDADeviceContext, double>);