Browse Source

add force infershape for some op

tags/v1.2.0
wxl 3 years ago
parent
commit
1227e0339f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/executor/worker/shape_inference_engine.cc

+ 1
- 1
ge/hybrid/executor/worker/shape_inference_engine.cc View File

@@ -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;
}



Loading…
Cancel
Save