提交 b639a882 编写于 作者: Z Zeng Jinle 提交者: gongweibao

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

上级 4d0f5ab1
......@@ -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);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册