Browse Source

ReshapeRemovePass_not_remove_reshape_in_dynamic_subgraph

pull/1883/head
lichun 4 years ago
parent
commit
d9916526de
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/reshape_remove_pass.cc

+ 1
- 1
ge/graph/passes/reshape_remove_pass.cc View File

@@ -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) {


Loading…
Cancel
Save