Browse Source

!477 modfiy mark agnostic pass about refmerge and refswitch

From: @zhou_chao1993
Reviewed-by: @wqtshg,@ljl0711
Signed-off-by: @ljl0711
tags/v1.1.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
e4636cef1b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/mark_agnostic_pass.cc

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

@@ -43,7 +43,7 @@ Status MarkAgnosticPass::Run(ComputeGraphPtr graph) {
if (node_type == REFMERGE || node_type == REFSWITCH) {
GELOGD("Mark format agnostic for regmerge and refswitch node %s", node->GetName().c_str());
AttrUtils::SetInt(node->GetOpDesc(), "_format_agnostic", 1);
AttrUtils::SetListInt(node->GetOpDesc(), "_format_agnostic_except_output", std::vector<int64_t>({1}));
AttrUtils::SetListInt(node->GetOpDesc(), "_format_agnostic_except_input", std::vector<int64_t>({1}));
continue;
}
if (node_type == MERGE) {


Loading…
Cancel
Save