Skip to content

Commit a1d3db0

Browse files
authored
Merge pull request #13844 from jerrywgz/fix_roi_pool
fix roi pool register
2 parents 0a9f5f1 + 5533426 commit a1d3db0

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/operators/roi_pool_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,4 +174,4 @@ REGISTER_OP_CPU_KERNEL(
174174
REGISTER_OP_CPU_KERNEL(
175175
roi_pool_grad,
176176
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, float>,
177-
ops::CPUROIPoolOpKernel<paddle::platform::CPUDeviceContext, double>);
177+
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>);

paddle/fluid/operators/roi_pool_op.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,4 +249,4 @@ REGISTER_OP_CUDA_KERNEL(
249249
REGISTER_OP_CUDA_KERNEL(
250250
roi_pool_grad,
251251
ops::GPUROIPoolGradOpKernel<paddle::platform::CUDADeviceContext, float>,
252-
ops::GPUROIPoolOpKernel<paddle::platform::CUDADeviceContext, double>);
252+
ops::GPUROIPoolGradOpKernel<paddle::platform::CUDADeviceContext, double>);

0 commit comments

Comments
 (0)