From 5478e1a532a2f9a7d585717a6d29e6425af4bf29 Mon Sep 17 00:00:00 2001
From: Ruibiao Chen <chenruibiao@baidu.com>
Date: Mon, 14 Nov 2022 11:15:34 +0800
Subject: [PATCH] Add InferShape for Depend OP (#47907)

---
 paddle/fluid/operators/controlflow/depend_op.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/paddle/fluid/operators/controlflow/depend_op.cc b/paddle/fluid/operators/controlflow/depend_op.cc
index e66bd33bbf..a4db1501fd 100644
--- a/paddle/fluid/operators/controlflow/depend_op.cc
+++ b/paddle/fluid/operators/controlflow/depend_op.cc
@@ -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);
-- 
GitLab