Browse Source

fix gen_mask control-edges bug

tags/v1.3.0
dingshihao2 3 years ago
parent
commit
8858e22807
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      ge/graph/passes/link_gen_mask_nodes_pass.cc

+ 2
- 0
ge/graph/passes/link_gen_mask_nodes_pass.cc View File

@@ -111,6 +111,8 @@ void LinkGenMaskNodesPass::GetAllGenMaskNodes(ComputeGraphPtr graph, vector<Node
// node gen_mask is located at different place in the fused node // node gen_mask is located at different place in the fused node
if (in_data_node->GetName().find(DROPOUTGENMASK) != in_data_node->GetName().npos) { if (in_data_node->GetName().find(DROPOUTGENMASK) != in_data_node->GetName().npos) {
gen_mask = in_data_node; gen_mask = in_data_node;
GELOGD("The fused node type [%s], paired with input node name [%s].", node->GetType(), gen_mask->GetName());
break;
} }
} }




Loading…
Cancel
Save