Searched refs:kPrimSGD (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.h | 51 …ckPrimitiveType(node, prim::kPrimApplyMomentum) || opt::CheckPrimitiveType(node, prim::kPrimSGD) || in IsOptimizerNode()
|
| D | to_format_base.cc | 273 …pt::CheckPrimitiveType(cnode, prim::kPrimAdam) || opt::CheckPrimitiveType(cnode, prim::kPrimSGD)) { in HandleGraphNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 276 if (opt::CheckPrimitiveType(cnode, prim::kPrimSGD)) { in DoOptimizerQuantize()
446 …:CheckPrimitiveType(cnode, prim::kPrimAdam)) || (opt::CheckPrimitiveType(cnode, prim::kPrimSGD)) || in DoQuantize()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 165 {prim::kPrimSGD, {InferImplSGD, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /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 | 391 inline const PrimitivePtr kPrimSGD = std::make_shared<Primitive>("SGD"); variable
|