Searched refs:equiv2 (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | lamb_next_mv_with_decay_rule.cc | 81 bool LambNextMVWithDecayRule::IsShareNodes(const EquivPtr &equiv1, const EquivPtr &equiv2) const { in IsShareNodes()
82 return IsSameNode(equiv1, equiv2, mul4_var_) && IsSameNode(equiv1, equiv2, real_div0_var_) && in IsShareNodes()
83 IsSameNode(equiv1, equiv2, real_div1_var_) && IsSameNode(equiv1, equiv2, constant_add2_y_); in IsShareNodes()
|
| D | lamb_next_mv_rule.cc | 101 bool LambNextMVRule::IsShareNodes(const EquivPtr &equiv1, const EquivPtr &equiv2) const { in IsShareNodes()
102 return IsSameNode(equiv1, equiv2, real_div0_var_) && IsSameNode(equiv1, equiv2, real_div1_var_) && in IsShareNodes()
103 IsSameNode(equiv1, equiv2, add2_y_); in IsShareNodes()
|
| D | lamb_next_mv_with_decay_rule.h | 49 bool IsShareNodes(const EquivPtr &equiv1, const EquivPtr &equiv2) const override;
|
| D | lamb_next_mv_rule.h | 59 bool IsShareNodes(const EquivPtr &equiv1, const EquivPtr &equiv2) const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | optimizer.h | 63 virtual bool IsShareNodes(const EquivPtr &equiv1, const EquivPtr &equiv2) const = 0;
|
| D | helper.h | 202 bool IsSameNode(const EquivPtr &equiv1, const EquivPtr &equiv2, const VarPtr &var_node);
|
| D | helper.cc | 830 bool IsSameNode(const EquivPtr &equiv1, const EquivPtr &equiv2, const VarPtr &var_node) { in IsSameNode() argument
832 MS_EXCEPTION_IF_NULL(equiv2); in IsSameNode()
836 auto equiv2_node = GetAnfNodeByVar(equiv2, var_node); in IsSameNode()
|