Browse Source

Feature: reset shape of dynamic single op

pull/565/head
l00444296 4 years ago
parent
commit
1d48442259
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      ge/graph/manager/graph_manager.cc

+ 3
- 0
ge/graph/manager/graph_manager.cc View File

@@ -2281,6 +2281,9 @@ Status GraphManager::OptimizeStage2(ge::ComputeGraphPtr &compute_graph) {
new (std::nothrow) VariableRefDeleteOpPass))
GE_CHK_STATUS_RET(pass_for_control_attr_optimize.AddPass("OptimizeStage2::ControlAttrOptimize::CompileNodesPass",
new (std::nothrow) CompileNodesPass))
GE_CHK_STATUS_RET(
pass_for_control_attr_optimize.AddPass("OptimizeStage2::AfterMergePasses::DynamicSingleOpResetShapePass",
new (std::nothrow) DynamicSingleOpResetShapePass))
GE_CHK_STATUS_RET(pass_for_control_attr_optimize.AddPass(
"OptimizeStage2::AfterMergePasses::MarkGraphUnknownStatusPass", new(std::nothrow) MarkGraphUnknownStatusPass))
GE_CHK_STATUS_RET(


Loading…
Cancel
Save