|
|
@@ -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; |
|
|
|
} |
|
|
|