Searched refs:kPrimApplyFtrl (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
D | optimizer_unify_output.cc | 74 VectorRef pattern({prim::kPrimApplyFtrl, var, accum, linear, grad, lr, l1, l2, lr_power, u}); in DefinePattern()
|
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
D | less_batch_normalization.cc | 252 prim::kPrimApplyFtrl, prim::kPrimSGD, prim::kPrimApplyRMSProp, prim::kPrimAdam};
|
/third_party/mindspore/mindspore/core/base/ |
D | core_ops.h | 346 inline const PrimitivePtr kPrimApplyFtrl = std::make_shared<Primitive>("ApplyFtrl"); variable
|