From 7098f98f6314a70f34e13c5aa5910f5639e224d2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=96=9B=E9=B9=8F?= Date: Sat, 25 Jun 2022 11:42:17 +0000 Subject: [PATCH] =?UTF-8?q?!576=20=E4=BF=AE=E5=A4=8D=E4=BF=9D=E5=AD=98dump?= =?UTF-8?q?=E8=8A=82=E7=82=B9=E5=90=8D=E6=97=B6=E7=9A=84bug=20Merge=20pull?= =?UTF-8?q?=20request=20!576=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 | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index 951101e..c38763e 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -206,6 +206,14 @@ void AddDumpOriginName(const std::string& subgraph_name, const ge::NodePtr paren } GELOGD("Add dump origin name %s for node %s.", original_names[0].c_str(), node->GetName().c_str()); } +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()); + } + GELOGD("Add dump origin name %s for node %s.", node->GetName().c_str(), node->GetName().c_str()); + } +} } // namespace ge namespace ge { @@ -273,6 +281,7 @@ Status GenSubgraphParseTasks(const ge::ComputeGraphPtr &parent_graph, std::deque Status PostOpProcessForSubgraph(const ParseArg &arg) { if (arg.parent_node == nullptr) { + AddDumpOriginNameForRootGraph(arg.graph); return SUCCESS; } std::string op_type = arg.parent_node->GetType();