未验证 提交 5478e1a5 编写于 作者: R Ruibiao Chen 提交者: GitHub

Add InferShape for Depend OP (#47907)

上级 c5e1011b
......@@ -59,6 +59,11 @@ class DependOp : public framework::OperatorBase {
}
};
class DependOpShapeInference : public framework::InferShapeBase {
public:
void operator()(framework::InferShapeContext *ctx) const override {}
};
class DependOpProtoMaker : public framework::OpProtoAndCheckerMaker {
public:
void Make() override {
......@@ -97,4 +102,5 @@ REGISTER_OPERATOR(
paddle::framework::EmptyGradOpMaker<paddle::framework::OpDesc>,
paddle::framework::EmptyGradOpMaker<paddle::imperative::OpBase>,
ops::DependOpProtoMaker,
ops::DependOpShapeInference,
ops::DependNoNeedBufferVarsInferer);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册