From 1227e0339ffd7ef7855c9d6b791a4926ce32d8b5 Mon Sep 17 00:00:00 2001 From: wxl Date: Thu, 11 Mar 2021 10:35:53 +0800 Subject: [PATCH] add force infershape for some op --- ge/hybrid/executor/worker/shape_inference_engine.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/hybrid/executor/worker/shape_inference_engine.cc b/ge/hybrid/executor/worker/shape_inference_engine.cc index 0a7f3985..27919589 100755 --- a/ge/hybrid/executor/worker/shape_inference_engine.cc +++ b/ge/hybrid/executor/worker/shape_inference_engine.cc @@ -41,7 +41,7 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) { // Wait for "const input nodes" if node's shape inference function requires any. // Even if output shape is static, there are cases that the const-input will be used in OpTiling and Execution GE_CHK_STATUS_RET_NOLOG(AwaitDependentNodes(node_state)); - if (node_item.is_output_shape_static && node_item.is_need_force_infershape) { + if (node_item.is_output_shape_static && !node_item.is_need_force_infershape) { return SUCCESS; }