未验证 提交 1c4134f6 编写于 作者: H HongyuJia 提交者: GitHub

polish REGISTER_OPERATOR parameter of fill_any (#45263)

上级 6d4ae007
......@@ -68,10 +68,10 @@ DECLARE_INPLACE_OP_INFERER(FillAnyGradInplaceInferer,
namespace ops = paddle::operators;
DECLARE_INFER_SHAPE_FUNCTOR(fill_any,
FillInferShapeFunctor,
FillAnyInferShapeFunctor,
PD_INFER_META(phi::UnchangedInferMeta));
DECLARE_INFER_SHAPE_FUNCTOR(fill_any_grad,
FillAnyInferShapeFunctor,
FillAnyGradInferShapeFunctor,
PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(fill_any,
......@@ -80,9 +80,9 @@ REGISTER_OPERATOR(fill_any,
ops::FillAnyGradOpMaker<paddle::framework::OpDesc>,
ops::FillAnyGradOpMaker<paddle::imperative::OpBase>,
ops::FillAnyOpInplaceInferer,
FillInferShapeFunctor);
FillAnyInferShapeFunctor);
REGISTER_OPERATOR(fill_any_grad,
ops::FillAnyGradOp,
ops::FillAnyGradInplaceInferer,
FillAnyInferShapeFunctor);
FillAnyGradInferShapeFunctor);
......@@ -51,8 +51,8 @@ const std::unordered_set<std::string> deprecated_op_names(
"squeeze",
"squeeze_grad",
"isfinite",
"matmul",
"fill",
"matmul",
"matmul_grad",
"matmul_grad_grad",
"max",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册