From 13dd31f56ca919872b6fcb5cc8dd5404967a9cb7 Mon Sep 17 00:00:00 2001 From: wilfChen Date: Fri, 4 Sep 2020 09:12:13 +0800 Subject: [PATCH] reorder fused optimizer --- mindspore/ccsrc/utils/utils.h | 4 ++++ 1 file changed, 4 insertions(+) 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, };