diff --git a/paddle/operators/get_places_op.cc b/paddle/operators/get_places_op.cc index 959aae16a8da43f2f9a0aebd37061dc5d60a9475..4d5059c264ec707fb6d9f14d680921bcfec8dbfc 100644 --- a/paddle/operators/get_places_op.cc +++ b/paddle/operators/get_places_op.cc @@ -58,8 +58,7 @@ class GetPlacesOp : public framework::OperatorBase { class GetPlacesOpProtoMaker : public framework::OpProtoAndCheckerMaker { public: - GetPlacesOpProtoMaker(framework::OpProto *proto, - framework::OpAttrChecker *op_checker) + GetPlacesOpProtoMaker(OpProto *proto, OpAttrChecker *op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { AddOutput("Out", "vector of Place"); AddAttr("device_count", "(int)device count").SetDefault(1);