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

Add InferShape for Depend OP (#47907)

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