ssk015
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix reshape reshape case in auto parallel

This commit is contained in:
yao_yf 2020-05-28 14:26:12 +08:00
parent 0f22140331
commit dcb91b0ef6
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/optimizer/irpass/reshape_eliminate.h
View File

@ -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;
}