File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -174,4 +174,4 @@ REGISTER_OP_CPU_KERNEL(
174
174
REGISTER_OP_CPU_KERNEL (
175
175
roi_pool_grad,
176
176
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, float >,
177
- ops::CPUROIPoolOpKernel <paddle::platform::CPUDeviceContext, double >);
177
+ ops::CPUROIPoolGradOpKernel <paddle::platform::CPUDeviceContext, double >);
Original file line number Diff line number Diff line change @@ -249,4 +249,4 @@ REGISTER_OP_CUDA_KERNEL(
249
249
REGISTER_OP_CUDA_KERNEL (
250
250
roi_pool_grad,
251
251
ops::GPUROIPoolGradOpKernel<paddle::platform::CUDADeviceContext, float >,
252
- ops::GPUROIPoolOpKernel <paddle::platform::CUDADeviceContext, double >);
252
+ ops::GPUROIPoolGradOpKernel <paddle::platform::CUDADeviceContext, double >);
You can’t perform that action at this time.
0 commit comments