From b639a882c3d20131131456f9ccf7292c39bf4fec Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Wed, 27 Nov 2019 10:15:31 +0800 Subject: [PATCH] fix syn bn grad maker, test=develop, test=document_fix (#21317) --- paddle/fluid/operators/sync_batch_norm_op.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/sync_batch_norm_op.cc b/paddle/fluid/operators/sync_batch_norm_op.cc index a134541dd7..7f441091d8 100644 --- a/paddle/fluid/operators/sync_batch_norm_op.cc +++ b/paddle/fluid/operators/sync_batch_norm_op.cc @@ -17,7 +17,7 @@ limitations under the License. */ namespace paddle { namespace operators { template -class BatchNormGradMaker : public framework::SingleGradOpMaker { +class SyncBatchNormGradMaker : public framework::SingleGradOpMaker { public: using framework::SingleGradOpMaker::SingleGradOpMaker; @@ -55,6 +55,6 @@ class BatchNormGradMaker : public framework::SingleGradOpMaker { namespace ops = paddle::operators; REGISTER_OPERATOR(sync_batch_norm, ops::BatchNormOp, ops::BatchNormOpMaker, ops::BatchNormOpInferVarType, - ops::BatchNormGradMaker, - ops::BatchNormGradMaker); + ops::SyncBatchNormGradMaker, + ops::SyncBatchNormGradMaker); REGISTER_OPERATOR(sync_batch_norm_grad, ops::BatchNormGradOp); -- GitLab