File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -11,6 +11,7 @@ limitations under the License. */
1111
1212#include " paddle/fluid/operators/pixel_shuffle_op.h"
1313#include < memory>
14+ #include " paddle/fluid/framework/op_version_registry.h"
1415
1516namespace paddle {
1617namespace operators {
@@ -185,3 +186,10 @@ REGISTER_OP_CPU_KERNEL(
185186 pixel_shuffle_grad,
186187 ops::PixelShuffleGradOpKernel<paddle::platform::CPUDeviceContext, float >,
187188 ops::PixelShuffleGradOpKernel<paddle::platform::CPUDeviceContext, double >);
189+
190+ REGISTER_OP_VERSION (pixel_shuffle)
191+ .AddCheckpoint(
192+ R"ROC(
193+ Compatible upgrade of pixel_shuffle, add a new attribute [data_format])ROC" ,
194+ paddle::framework::compatible::OpVersionDesc ().NewAttr(
195+ " data_format" , " Specify the data format of the input data" , true ));
You can’t perform that action at this time.
0 commit comments