add force infershape for some op

pull/1224/head
wxl 4 years ago
parent 5ae267433b
commit 1227e0339f

@ -41,7 +41,7 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) {
// Wait for "const input nodes" if node's shape inference function requires any. // 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 // 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)); 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; return SUCCESS;
} }

Loading…
Cancel
Save