Update logical_op.cc

test=develop
6 years ago · 916930a8ae
parent 0c233e8870
commit 916930a8ae
1 changed files with 1 additions and 1 deletions
--- 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,