Browse Source

Fix CSE bug for `DropoutGenMask` operation.

tags/v0.3.0-alpha
seatea 6 years ago
parent
commit
538cc7e21b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/optimizer/cse.cc

+ 3
- 0
mindspore/ccsrc/optimizer/cse.cc View File

@@ -122,6 +122,9 @@ bool CSE::CheckReplace(const AnfNodePtr &main, const AnfNodePtr &node) const {
break;
}
}
if (IsPrimitiveCNode(c_main, prim::kPrimDropoutGenMask)) {
appsame = false;
}
replace = appsame;
}
}


Loading…
Cancel
Save