fix syn bn grad maker, test=develop, test=document_fix (#21317)

revert-21172-masked_select_api
Zeng Jinle 5 years ago committed by gongweibao
parent 4d0f5ab1a8
commit b639a882c3

@ -17,7 +17,7 @@ limitations under the License. */
namespace paddle {
namespace operators {
template <typename T>
class BatchNormGradMaker : public framework::SingleGradOpMaker<T> {
class SyncBatchNormGradMaker : public framework::SingleGradOpMaker<T> {
public:
using framework::SingleGradOpMaker<T>::SingleGradOpMaker;
@ -55,6 +55,6 @@ class BatchNormGradMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators;
REGISTER_OPERATOR(sync_batch_norm, ops::BatchNormOp, ops::BatchNormOpMaker,
ops::BatchNormOpInferVarType,
ops::BatchNormGradMaker<paddle::framework::OpDesc>,
ops::BatchNormGradMaker<paddle::imperative::OpBase>);
ops::SyncBatchNormGradMaker<paddle::framework::OpDesc>,
ops::SyncBatchNormGradMaker<paddle::imperative::OpBase>);
REGISTER_OPERATOR(sync_batch_norm_grad, ops::BatchNormGradOp);

Loading…
Cancel
Save