diff --git a/paddle/fluid/operators/detection/box_clip_op.cc b/paddle/fluid/operators/detection/box_clip_op.cc index fb94d0fbc619f13e73da5608b9ed2980bb91b124..e47027d98c40c40b6b5b763942ac70d2dc4bb4b8 100644 --- a/paddle/fluid/operators/detection/box_clip_op.cc +++ b/paddle/fluid/operators/detection/box_clip_op.cc @@ -45,7 +45,7 @@ class BoxClipOp : public framework::OperatorWithKernel { protected: framework::OpKernelType GetExpectedKernelType( const framework::ExecutionContext& ctx) const override { - auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("RpnRois")); + auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("InputBox")); return framework::OpKernelType(data_type, platform::CPUPlace()); } };