Searched refs:kPrimApplyRMSProp (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | optimizer_unify_output.cc | 100 VectorRef pattern({prim::kPrimApplyRMSProp, inputs}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 77 … {prim::kPrimApplyRMSProp, {6, 7, 8}}, in InConvertWhiteList()
92 {prim::kPrimAssignSub, {1}}, {prim::kPrimApplyRMSProp, {6, 7, 8}}, in InConvertWhiteList()
|
| D | less_batch_normalization.cc | 252 prim::kPrimApplyFtrl, prim::kPrimSGD, prim::kPrimApplyRMSProp, prim::kPrimAdam};
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 386 inline const PrimitivePtr kPrimApplyRMSProp = std::make_shared<Primitive>("ApplyRMSProp"); variable
|