未验证 提交 844d8e0c 编写于 作者: W wangguanzhong 提交者: GitHub

add REGISTER_OP_VERSION for generate_proposals, roi_align, roi_pool test=op_version (#30034)

上级 c3c064a8
...@@ -303,6 +303,13 @@ REGISTER_OPERATOR( ...@@ -303,6 +303,13 @@ REGISTER_OPERATOR(
REGISTER_OP_CPU_KERNEL(generate_proposals, ops::GenerateProposalsKernel<float>, REGISTER_OP_CPU_KERNEL(generate_proposals, ops::GenerateProposalsKernel<float>,
ops::GenerateProposalsKernel<double>); ops::GenerateProposalsKernel<double>);
REGISTER_OP_VERSION(generate_proposals) REGISTER_OP_VERSION(generate_proposals)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of output [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteOutput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect output name and "
"it is not used in object detection models yet."))
.AddCheckpoint( .AddCheckpoint(
R"ROC( R"ROC(
Upgrade generate_proposals add a new output [RpnRoisNum])ROC", Upgrade generate_proposals add a new output [RpnRoisNum])ROC",
......
...@@ -233,6 +233,13 @@ REGISTER_OP_CPU_KERNEL( ...@@ -233,6 +233,13 @@ REGISTER_OP_CPU_KERNEL(
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, double>, ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, double>,
ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, int>); ops::CPUROIAlignGradOpKernel<paddle::platform::CPUDeviceContext, int>);
REGISTER_OP_VERSION(roi_align) REGISTER_OP_VERSION(roi_align)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of input [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteInput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect input name and "
"it is not used in object detection models yet."))
.AddCheckpoint( .AddCheckpoint(
R"ROC( R"ROC(
Upgrade roi_align add a new input [RoisNum])ROC", Upgrade roi_align add a new input [RoisNum])ROC",
......
...@@ -227,6 +227,13 @@ REGISTER_OP_CPU_KERNEL( ...@@ -227,6 +227,13 @@ REGISTER_OP_CPU_KERNEL(
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>, ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, double>,
ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, int>); ops::CPUROIPoolGradOpKernel<paddle::platform::CPUDeviceContext, int>);
REGISTER_OP_VERSION(roi_pool) REGISTER_OP_VERSION(roi_pool)
.AddCheckpoint(
R"ROC(
Incompatible upgrade of input [RpnRoisLod])ROC",
paddle::framework::compatible::OpVersionDesc().DeleteInput(
"RpnRoisLod",
"Delete RpnRoisLod due to incorrect input name and "
"it is not used in object detection models yet."))
.AddCheckpoint( .AddCheckpoint(
R"ROC( R"ROC(
Upgrade roi_pool add a new input [RoisNum])ROC", Upgrade roi_pool add a new input [RoisNum])ROC",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册