xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!14502 modify export mindir tuple[] 

From: @changzherui
Reviewed-by: @zhoufeng54,@zh_qh
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2021-04-01 15:57:00 +08:00 committed by Gitee
parent 18d79d35b6 c7323b39b3
commit ae91575346
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/transform/express_ir/mindir_exporter.cc
View File

@ -443,6 +443,7 @@ std::string IrExportBuilder::BuildInputNode(const AnfNodePtr &node, mind_ir::Gra
if (node->isa<ValueNode>()) {
// 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);
}

8
mindspore/core/load_mindir/anf_model_parser.cc
View File

@ -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<ValuePtr>{});
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;
}