提交 2fe99f2b 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5740 Reorder gpu fused optimizer

Merge pull request !5740 from chenweifeng/gpu-momentum-fusion
......@@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
kApplyProximalAdagradOpName,
kApplyProximalGradientDescentOpName,
kApplyRMSPropOpName,
kFusedAdamWeightDecayName,
kFusedAdamName,
kFusedWeightScaleApplyMomentum,
kFusedScaleApplyMomentum,
kPullOpName,
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册