diff --git a/mindspore/ccsrc/utils/utils.h b/mindspore/ccsrc/utils/utils.h index ba6cd9c4127..5a395eb018b 100644 --- a/mindspore/ccsrc/utils/utils.h +++ b/mindspore/ccsrc/utils/utils.h @@ -344,6 +344,10 @@ const std::set kOptOperatorSet = { kApplyProximalAdagradOpName, kApplyProximalGradientDescentOpName, kApplyRMSPropOpName, + kFusedAdamWeightDecayName, + kFusedAdamName, + kFusedWeightScaleApplyMomentum, + kFusedScaleApplyMomentum, kPullOpName, };