Skip to content

Commit

Permalink
Update pixel_unshuffle_op.cc
Browse files Browse the repository at this point in the history
  • Loading branch information
BrilliantYuKaimin committed Apr 21, 2022
1 parent e871227 commit 948f32b
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions paddle/fluid/operators/pixel_unshuffle_op.cc
Expand Up @@ -16,8 +16,8 @@
#include "paddle/fluid/framework/op_registry.h"
#include "paddle/fluid/framework/op_version_registry.h"
#include "paddle/phi/core/infermeta_utils.h"
#include "paddle/phi/infermeta/unary.h"
#include "paddle/phi/infermeta/backward.h"
#include "paddle/phi/infermeta/unary.h"

namespace paddle {
namespace operators {
Expand Down Expand Up @@ -99,6 +99,5 @@ DECLARE_INFER_SHAPE_FUNCTOR(pixel_unshuffle_grad,
PixelUnshuffleGradInferShapeFunctor,
PD_INFER_META(phi::PixelUnshuffleGradInferMeta));

REGISTER_OPERATOR(pixel_unshuffle_grad,
ops::PixelUnshuffleGradOp,
REGISTER_OPERATOR(pixel_unshuffle_grad, ops::PixelUnshuffleGradOp,
PixelUnshuffleGradInferShapeFunctor);

0 comments on commit 948f32b

Please sign in to comment.