diff --git a/mindspore/lite/src/common/graph_util.cc b/mindspore/lite/src/common/graph_util.cc index c8472bd7f9..e094c744ab 100755 --- a/mindspore/lite/src/common/graph_util.cc +++ b/mindspore/lite/src/common/graph_util.cc @@ -34,7 +34,9 @@ std::vector GetGraphInputNodes(const schema::MetaGraph *meta_graph) { MS_ASSERT(nullptr != cNode); for (size_t k = 0; k < cNode->inputIndex()->size(); k++) { if (cNode->inputIndex()->GetAs(k) == input_index) { - ret.emplace_back(j); + if (!IsContain(ret, j)) { + ret.emplace_back(j); + } break; } } @@ -53,7 +55,9 @@ std::vector GetGraphOutputNodes(const schema::MetaGraph *meta_graph) { MS_ASSERT(nullptr != cNode); for (size_t k = 0; k < cNode->outputIndex()->size(); k++) { if (cNode->outputIndex()->GetAs(k) == output_index) { - ret.emplace_back(j); + if (!IsContain(ret, j)) { + ret.emplace_back(j); + } break; } }