Browse Source

fix gen_mask control-edges bug

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

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

@@ -111,7 +111,8 @@ void LinkGenMaskNodesPass::GetAllGenMaskNodes(ComputeGraphPtr graph, vector<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) {
gen_mask = in_data_node;
GELOGD("The fused node type [%s], paired with input node name [%s].", node->GetType(), gen_mask->GetName());
GELOGD("The fused node type [%s], paired with the input node name [%s].",
node->GetType().c_str(), gen_mask->GetName().c_str());
break;
}
}


Loading…
Cancel
Save