Searched refs:kPrimAdam (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.h | 52 opt::CheckPrimitiveType(node, prim::kPrimAdam); in IsOptimizerNode()
|
| D | to_format_base.cc | 273 …if (opt::CheckPrimitiveType(cnode, prim::kPrimAdam) || opt::CheckPrimitiveType(cnode, prim::kPrimS… in HandleGraphNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 273 if (opt::CheckPrimitiveType(cnode, prim::kPrimAdam)) { in DoOptimizerQuantize()
446 …} else if ((opt::CheckPrimitiveType(cnode, prim::kPrimAdam)) || (opt::CheckPrimitiveType(cnode, pr… in DoQuantize()
|
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | node_util.cc | 338 …rimitiveType(cnode, prim::kPrimConv2DFusion) || opt::CheckPrimitiveType(cnode, prim::kPrimAdam))) { in GetCNodeOutputsSize()
|
| /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/lite/tools/optimizer/graph/ |
| D | node_infershape.cc | 391 …CheckPrimitiveType(cnode, prim::kPrimConv2DFusion) || CheckPrimitiveType(cnode, prim::kPrimAdam)) { in GetCNodeOutputTensors()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 260 inline const PrimitivePtr kPrimAdam = std::make_shared<Primitive>("Adam"); variable
|