From d6404c9a8f3c0ca1276c8f65f7c25b1c009695b4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=96=9B=E9=B9=8F?= Date: Mon, 27 Jun 2022 11:00:17 +0000 Subject: [PATCH] =?UTF-8?q?!580=20bugfix=20for=20code=20check=20Merge=20pu?= =?UTF-8?q?ll=20request=20!580=20from=20=E8=96=9B=E9=B9=8F/bugfix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- parser/tensorflow/tensorflow_parser.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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