diff --git a/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc b/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc index 78e2d3e5fa..82708121d6 100644 --- a/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc +++ b/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc @@ -177,7 +177,8 @@ void FusedBatchNormFusion::EliminateMonadNodes(const FuncGraphPtr &func_graph, c MS_EXCEPTION_IF_NULL(manager); auto assign_sub1 = GetAnfNodeByVar(equiv, assign_sub1_var_); MS_EXCEPTION_IF_NULL(assign_sub1); - for (const auto &node_index : manager->node_users()[assign_sub1]) { + auto users = manager->node_users()[assign_sub1]; + for (const auto &node_index : users) { const AnfNodePtr &output = node_index.first; MS_EXCEPTION_IF_NULL(output); if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimUpdateState)) {