Browse Source

reorder fused optimizer

tags/v1.0.0
wilfChen 5 years ago
parent
commit
13dd31f56c
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/utils/utils.h

+ 4
- 0
mindspore/ccsrc/utils/utils.h View File

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




Loading…
Cancel
Save