Searched refs:kPrimApplyCenteredRMSProp (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
D | optimizer_unify_output.cc | 120 …VectorRef pattern({prim::kPrimApplyCenteredRMSProp, var, mg, ms, mom, grad, lr, rho, momentum, eps… in DefinePattern()
|
/third_party/mindspore/mindspore/core/base/ |
D | core_ops.h | 288 inline const PrimitivePtr kPrimApplyCenteredRMSProp = std::make_shared<Primitive>("ApplyCenteredRMS… variable
|
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | anf_runtime_algorithm.cc | 167 {prim::kPrimApplyCenteredRMSProp->name(),
|