Searched refs:kPrimApplyMomentum (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | replace_momentum_cast_fusion.cc | 26 VectorRef momentum = VectorRef({prim::kPrimApplyMomentum, var_, acc_, lr_, grad_cast, mom_}); in DefinePattern()
|
| D | apply_momentum_weight_fusion.cc | 32 …{prim::kPrimApplyMomentum, variable_, accumulation_, learning_rate_, weight_decay, momentum_, mona… in DefinePattern()
|
| D | apply_momentum_scale_fusion.cc | 60 …VectorRef({prim::kPrimApplyMomentum, variable_, accumulation_, learning_rate_, scale, momentum_, m… in DefinePattern()
|
| D | apply_momentum_weight_scale_fusion.cc | 92 …VectorRef({prim::kPrimApplyMomentum, variable_, accumulation_, learning_rate_, scale, momentum_, m… in DefinePattern()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.h | 51 …return opt::CheckPrimitiveType(node, prim::kPrimApplyMomentum) || opt::CheckPrimitiveType(node, pr… in IsOptimizerNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | momentum_lossscale_fusion.cc | 49 return VectorRef({prim::kPrimApplyMomentum, X0, X1, X2, VectorRef({prim::kPrimMul, Xs}), X4, X5}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | optimizer_unify_output.cc | 89 VectorRef pattern({prim::kPrimApplyMomentum, var, accum, lr, grad, momentum, u}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 57 …std::vector<std::pair<PrimitivePtr, std::vector<size_t>>> white_list({{prim::kPrimApplyMomentum, {… in InConvertWhiteList()
84 {{prim::kPrimApplyMomentum, {1, 2}}, {prim::kPrimMomentum, {2, 3}}, in InConvertWhiteList()
|
| D | less_batch_normalization.cc | 251 prim::kPrimLoad, prim::kPrimRefToEmbed, prim::kPrimApplyMomentum, prim::kPrimMomentum,
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 447 (opt::CheckPrimitiveType(cnode, prim::kPrimApplyMomentum))) { in DoQuantize()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 345 inline const PrimitivePtr kPrimApplyMomentum = std::make_shared<Primitive>("ApplyMomentum"); variable
|