From c7323b39b33301b5e22990e4e832ea30431359fe Mon Sep 17 00:00:00 2001 From: changzherui Date: Wed, 31 Mar 2021 19:56:42 +0800 Subject: [PATCH] modify export mindir tuple[] --- mindspore/ccsrc/transform/express_ir/mindir_exporter.cc | 1 + mindspore/core/load_mindir/anf_model_parser.cc | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/mindspore/ccsrc/transform/express_ir/mindir_exporter.cc b/mindspore/ccsrc/transform/express_ir/mindir_exporter.cc index 9e7a121aac..cb487a5fe6 100644 --- a/mindspore/ccsrc/transform/express_ir/mindir_exporter.cc +++ b/mindspore/ccsrc/transform/express_ir/mindir_exporter.cc @@ -443,6 +443,7 @@ std::string IrExportBuilder::BuildInputNode(const AnfNodePtr &node, mind_ir::Gra if (node->isa()) { // When node input is a ValueNode, need to create a Constant Node mind_ir::NodeProto *node_proto = graph_proto->add_node(); + node_proto->set_name(node_name); node_proto->add_output(node_name); SetAttributeProto(node, node_proto); } diff --git a/mindspore/core/load_mindir/anf_model_parser.cc b/mindspore/core/load_mindir/anf_model_parser.cc index f5b4f09b3e..d65e59276c 100644 --- a/mindspore/core/load_mindir/anf_model_parser.cc +++ b/mindspore/core/load_mindir/anf_model_parser.cc @@ -637,6 +637,14 @@ bool MSANFModelParser::GetAttrValueForValueNode(const std::string &value_node_na anfnode_build_map_[value_node_name] = new_value_node; break; } + if ((value_pos = ref_attr_name.find("Tuple[]")) != std::string::npos) { + MS_LOG(INFO) << "Build Tuple() ValueNode for primitive."; + ValuePtr res = MakeValue(std::vector{}); + new_value_node = NewValueNode(res); + new_value_node->set_abstract(res->ToAbstract()); + anfnode_build_map_[value_node_name] = new_value_node; + break; + } ObtainCNodeAttrInScalarForm(attr_proto, &multi_value_map); break; }