reorder fused optimizer

This commit is contained in:
wilfChen 2020-09-04 09:12:13 +08:00
parent 722bcc2323
commit 13dd31f56c
1 changed files with 4 additions and 0 deletions
mindspore/ccsrc/utils

View File

@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
kApplyProximalAdagradOpName, kApplyProximalAdagradOpName,
kApplyProximalGradientDescentOpName, kApplyProximalGradientDescentOpName,
kApplyRMSPropOpName, kApplyRMSPropOpName,
kFusedAdamWeightDecayName,
kFusedAdamName,
kFusedWeightScaleApplyMomentum,
kFusedScaleApplyMomentum,
kPullOpName, kPullOpName,
}; };