!477 modfiy mark agnostic pass about refmerge and refswitch

From: @zhou_chao1993
Reviewed-by: @wqtshg,@ljl0711
Signed-off-by: @ljl0711
pull/477/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e4636cef1b

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

Loading…
Cancel
Save