modify load mindir none type

4 years ago · 131a1bbb23
parent d0bd77c1ef
commit 131a1bbb23
1 changed files with 1 additions and 0 deletions
--- a/mindspore/core/load_mindir/anf_model_parser.cc
+++ b/mindspore/core/load_mindir/anf_model_parser.cc
@ -560,6 +560,7 @@ bool MSANFModelParser::ObtainValueNodeInNoneForm(const std::string &value_node_n
                                                 const mind_ir::AttributeProto &attr_proto) {
  auto new_value_node = NewValueNode(kNone);
  MS_EXCEPTION_IF_NULL(new_value_node);
+  new_value_node->set_abstract(kNone->ToAbstract());
  anfnode_build_map_[value_node_name] = new_value_node;
  return true;
 }