diff --git a/paddle/fluid/operators/controlflow/logical_op.cc b/paddle/fluid/operators/controlflow/logical_op.cc index 6d8a20e385b2585eea653d75c2f3b73e7cc3c7d0..cc28ffd6357336260110eade9d1f5c2eff3841ac 100644 --- a/paddle/fluid/operators/controlflow/logical_op.cc +++ b/paddle/fluid/operators/controlflow/logical_op.cc @@ -74,7 +74,7 @@ class BinaryLogicalOpInferShape : public framework::InferShapeBase { int product_x = framework::product(dim_x); int product_y = framework::product(dim_y); - bool check = ctx->IsRuntime() && product_x >= 0 && product_y >= 0; + bool check = ctx->IsRuntime() || (product_x >= 0 && product_y >= 0); if (check) { PADDLE_ENFORCE_EQ( product_x, product_y,