forked from mindspore-Ecosystem/mindspore
reorder fused optimizer
This commit is contained in:
parent
722bcc2323
commit
13dd31f56c
mindspore/ccsrc/utils
|
@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
|
||||||
kApplyProximalAdagradOpName,
|
kApplyProximalAdagradOpName,
|
||||||
kApplyProximalGradientDescentOpName,
|
kApplyProximalGradientDescentOpName,
|
||||||
kApplyRMSPropOpName,
|
kApplyRMSPropOpName,
|
||||||
|
kFusedAdamWeightDecayName,
|
||||||
|
kFusedAdamName,
|
||||||
|
kFusedWeightScaleApplyMomentum,
|
||||||
|
kFusedScaleApplyMomentum,
|
||||||
kPullOpName,
|
kPullOpName,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue