Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Doptimizer_unify_output.cc120 …VectorRef pattern({prim::kPrimApplyCenteredRMSProp, var, mg, ms, mom, grad, lr, rho, momentum, eps… in DefinePattern()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h288 inline const PrimitivePtr kPrimApplyCenteredRMSProp = std::make_shared<Primitive>("ApplyCenteredRMS… variable
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc167 {prim::kPrimApplyCenteredRMSProp->name(),