diff --git a/paddle/fluid/operators/grid_sampler_op.cc b/paddle/fluid/operators/grid_sampler_op.cc index 3d34a3d15c1ddd944dd205def278beeeef3efdeb..e357133be440d0d396a1cdc580127bf47dce6841 100644 --- a/paddle/fluid/operators/grid_sampler_op.cc +++ b/paddle/fluid/operators/grid_sampler_op.cc @@ -16,6 +16,7 @@ limitations under the License. */ #include #include #include "paddle/fluid/framework/op_registry.h" +#include "paddle/fluid/framework/op_version_registry.h" #ifdef PADDLE_WITH_CUDA #include "paddle/fluid/platform/cudnn_helper.h" #endif @@ -237,3 +238,11 @@ REGISTER_OP_CPU_KERNEL( grid_sampler_grad, ops::GridSampleGradOpKernel, ops::GridSampleGradOpKernel); + +REGISTER_OP_VERSION(grid_sampler) + .AddCheckpoint( + R"ROC( + Upgrade grid_sampler add a new attribute [mode]. + )ROC", + paddle::framework::compatible::OpVersionDesc().NewAttr( + "mode", "In order to specify interpolation mode", "bilinear"));