From 29edcfa0b66a3b50d4a6fa7692236398b4a062cb Mon Sep 17 00:00:00 2001 From: zhou_chao1993 Date: Tue, 1 Dec 2020 19:58:01 +0800 Subject: [PATCH] revert code --- ge/graph/passes/mark_agnostic_pass.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/passes/mark_agnostic_pass.cc b/ge/graph/passes/mark_agnostic_pass.cc index be8d132a..8c9a0451 100644 --- a/ge/graph/passes/mark_agnostic_pass.cc +++ b/ge/graph/passes/mark_agnostic_pass.cc @@ -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({1})); + AttrUtils::SetListInt(node->GetOpDesc(), "_format_agnostic_except_input", std::vector({1})); continue; } if (node_type == MERGE) {