diff --git a/paddle/operators/norm_op.cc b/paddle/operators/norm_op.cc index d23805da8612ae1f28a0c535e428e27478ea1036..990a1504ea34e7422518fb708a1ee920f30daadc 100644 --- a/paddle/operators/norm_op.cc +++ b/paddle/operators/norm_op.cc @@ -18,7 +18,7 @@ namespace operators { class NormOpMaker : public framework::OpProtoAndCheckerMaker { public: - NormOpMaker(framework::OpProto* proto, framework::OpAttrChecker* op_checker) + NormOpMaker(OpProto* proto, OpAttrChecker* op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { AddInput( "X", diff --git a/python/paddle/v2/fluid/tests/test_norm_op.py b/python/paddle/v2/fluid/tests/test_norm_op.py index 23e6841b916a6a26f3c1b4ec4a2569e70e2fdbf9..7d56320489b24c5547e045cb51b778851ff94a32 100644 --- a/python/paddle/v2/fluid/tests/test_norm_op.py +++ b/python/paddle/v2/fluid/tests/test_norm_op.py @@ -49,7 +49,7 @@ class TestNormOp(OpTest): self.check_grad(['X'], 'Out') def init_test_case(self): - self.shape = [1, 3, 2, 2] + self.shape = [2, 3, 2, 2] self.epsilon = 1e-6