Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcombine_momentum_fusion.cc30 if (AnfAlgo::GetCNodeName(momentum_node) == kFusedScaleApplyMomentum) { in GetDealList()
64 if (AnfAlgo::GetCNodeName(momentums[0]) == kFusedScaleApplyMomentum) { in Run()
Dapply_momentum_scale_fusion.cc84 auto prim = std::make_shared<Primitive>(kFusedScaleApplyMomentum); in Process()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h277 constexpr auto kFusedScaleApplyMomentum = "FusedScaleApplyMomentum"; variable
644 kFusedScaleApplyMomentum,