Browse Source

!580 bugfix for code check

Merge pull request !580 from 薛鹏/bugfix
pull/579/head
薛鹏 i-robot 3 years ago
parent
commit
d6404c9a8f
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      parser/tensorflow/tensorflow_parser.cc

+ 3
- 3
parser/tensorflow/tensorflow_parser.cc View File

@@ -208,10 +208,10 @@ void AddDumpOriginName(const std::string& subgraph_name, const ge::NodePtr paren
} }
void AddDumpOriginNameForRootGraph(const ge::ComputeGraphPtr& graph) { void AddDumpOriginNameForRootGraph(const ge::ComputeGraphPtr& graph) {
for (auto &node : graph->GetDirectNode()) { 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 } // namespace ge


Loading…
Cancel
Save