Home
last modified time | relevance | path

Searched refs:kFusedWeightScaleApplyMomentum (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcombine_momentum_fusion.cc32 } else if (AnfAlgo::GetCNodeName(momentum_node) == kFusedWeightScaleApplyMomentum) { in GetDealList()
Dapply_momentum_weight_scale_fusion.cc119 auto prim = std::make_shared<Primitive>(kFusedWeightScaleApplyMomentum); in Process()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h275 constexpr auto kFusedWeightScaleApplyMomentum = "FusedWeightScaleApplyMomentum"; variable
643 kFusedWeightScaleApplyMomentum,