未验证 提交 916930a8 编写于 作者: Y Yan Chunwei 提交者: GitHub

Update logical_op.cc

test=develop
上级 0c233e88
...@@ -74,7 +74,7 @@ class BinaryLogicalOpInferShape : public framework::InferShapeBase { ...@@ -74,7 +74,7 @@ class BinaryLogicalOpInferShape : public framework::InferShapeBase {
int product_x = framework::product(dim_x); int product_x = framework::product(dim_x);
int product_y = framework::product(dim_y); 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) { if (check) {
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
product_x, product_y, product_x, product_y,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册