Browse Source

!4644 Update value comparing with 0 in Greater

Merge pull request !4644 from thlinh/dev_Aug17_update_greater_check
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
a812bf2ef3
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/frontend/optimizer/irpass/value_based_eliminate.cc

+ 2
- 2
mindspore/ccsrc/frontend/optimizer/irpass/value_based_eliminate.cc View File

@@ -98,10 +98,10 @@ AnfNodePtr ValueBasedEliminate::operator()(const OptimizerPtr &, const AnfNodePt
PConstant zero_scalar_(node, false, 0, true);

// {prim::kPrimSelect, {prim::kPrimGreater, X, 0}, Y, Z}} -> Y when X is always greater than 0
MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreater, x, zero_), y, z), y,
MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreaterEqual, x, zero_), y, z), y,
IsCNodePositive(x.GetNode(node)));

MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreater, x, zero_scalar_), y, z), y,
MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreaterEqual, x, zero_scalar_), y, z), y,
IsCNodePositive(x.GetNode(node)));

// {prim::kPrimMaximum, X, LOWER_FLT_LIMIT}} -> X


Loading…
Cancel
Save