diff --git a/paddle/fluid/operators/sync_batch_norm_op.cc b/paddle/fluid/operators/sync_batch_norm_op.cc index a134541dd78066a73d813a6bd19b9ef65c0a48e7..7f441091d8b7c4c22b914f02380e744c69c93942 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);