From ae47ba2e22c32572953c9f28418100202631b78e Mon Sep 17 00:00:00 2001 From: y00500818 Date: Fri, 23 Jul 2021 15:59:00 +0800 Subject: [PATCH] bugfix for NormalizeInputOrOutputMap --- parser/tensorflow/tensorflow_parser.cc | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index 568d8a1..8fde8fc 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -2109,9 +2109,12 @@ Status TensorFlowModelParser::NormalizeInputOrOutputMap(const string &node_name, std::set 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; }