Skip to content

Commit 3c96333

Browse files
committed
fix roi pool register
1 parent e176170 commit 3c96333

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
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>);

0 commit comments

Comments
 (0)