diff --git a/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h b/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h index 5e7ce7cdd73d97c37c1402695e32d564b323dcdc..f1f73de4d930f46f84665f01b052e3c7f8b146ed 100644 --- a/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h +++ b/mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h @@ -88,7 +88,9 @@ class TwoReshapeEliminater : public AnfVisitor { auto fg = node->func_graph(); if (fg != nullptr && x_ != nullptr && shape_ != nullptr) { - return fg->NewCNode({NewValueNode(prim_), x_, shape_}); + auto new_node = fg->NewCNode({NewValueNode(prim_), x_, shape_}); + new_node->set_abstract(node->abstract()); + return new_node; } return nullptr; }