Browse Source

!348 bugfix for NormalizeInputOrOutputMap

Merge pull request !348 from yangyongqiang/master
r1.5.1
i-robot Gitee 3 years ago
parent
commit
d69ce1fce8
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      parser/tensorflow/tensorflow_parser.cc

+ 5
- 2
parser/tensorflow/tensorflow_parser.cc View File

@@ -2109,9 +2109,12 @@ Status TensorFlowModelParser::NormalizeInputOrOutputMap(const string &node_name,
std::set<std::string> compare_set;

for (auto &pair : pairs) {
bool is_fusion_child = (fusion_op_children_.find(node_name) != fusion_op_children_.end()) ||
(fusion_op_children_.find(iter->first) != fusion_op_children_.end());
bool is_fusion_op = (fusion_op_type_map_.find(node_name) != fusion_op_type_map_.end()) ||
(fusion_op_type_map_.find(iter->first) != fusion_op_type_map_.end());
if (((pair.first == ge::kFusionDisableIndex) || (pair.second == ge::kFusionDisableIndex)) &&
((fusion_op_children_.find(node_name) != fusion_op_children_.end()) ||
(fusion_op_children_.find(iter->first) != fusion_op_children_.end()))) {
(is_fusion_child || is_fusion_op)) {
// The edge will be cut off at the back, ignoring
continue;
}


Loading…
Cancel
Save