diff --git a/mindspore/ccsrc/utils/utils.h b/mindspore/ccsrc/utils/utils.h index ba6cd9c412736a6d492831ab2e5e9445f9a478f3..5a395eb018b9e7fc163dc2067027e6e7ecb142ae 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, };