diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index c38763e..a1a2d9a 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -208,10 +208,10 @@ void AddDumpOriginName(const std::string& subgraph_name, const ge::NodePtr paren } void AddDumpOriginNameForRootGraph(const ge::ComputeGraphPtr& graph) { for (auto &node : graph->GetDirectNode()) { - if (!ge::AttrUtils::SetListStr(node->GetOpDesc(), ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, {node->GetName()})) { - GELOGW("Set %s to %s fail.", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES.c_str(), node->GetOpDesc()->GetName().c_str()); + if (ge::AttrUtils::SetListStr(node->GetOpDesc(), ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, {node->GetName()})) { + GELOGD("Add dump origin name %s for node %s.", node->GetName().c_str(), + node->GetName().c_str()); } - GELOGD("Add dump origin name %s for node %s.", node->GetName().c_str(), node->GetName().c_str()); } } } // namespace ge