Home
last modified time | relevance | path

Searched refs:kApplyMomentumOpName (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dinput_to_output_registry.cc80 Register(kApplyMomentumOpName, {1}); in InputToOutputRegistry()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dcommon.h183 {kApplyMomentumOpName, kMomentumNameToIdx}, {kFusedSparseAdamName, kSparseAdamNameToIdx},
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h104 constexpr auto kApplyMomentumOpName = "ApplyMomentum"; variable
620 kApplyMomentumOpName,
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc466 if (AnfAlgo::GetCNodeName(kernel) == kApplyMomentumOpName) { in RunOpAssignOutputMemory()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc1792 if (AnfAlgo::GetCNodeName(child) == kApplyMomentumOpName) { in FindDelayExecPosition()