|
|
@@ -47,7 +47,7 @@ Status ReshapeRemovePass::Run(NodePtr &node) { |
|
|
case kReshapeType: { |
|
|
case kReshapeType: { |
|
|
bool is_in_unknown_shape_graph = false; |
|
|
bool is_in_unknown_shape_graph = false; |
|
|
bool forced_unknown = false; |
|
|
bool forced_unknown = false; |
|
|
for (const auto &node : graph->GetDirectNode()) { |
|
|
|
|
|
|
|
|
for (const auto &node : node->GetOwnerComputeGraph()->GetDirectNode()) { |
|
|
GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_shape_graph), |
|
|
GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_shape_graph), |
|
|
"[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str()); |
|
|
"[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str()); |
|
|
if (is_in_unknown_shape_graph) { |
|
|
if (is_in_unknown_shape_graph) { |
|
|
|