Searched refs:kPrimMomentum (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 67 …MATCH_REPLACE(node, PPrimitive(prim::kPrimMomentum, PPrimitive(prim::kPrimZerosLike, x), y, z).Min… in operator ()()
|
| D | branch_culling.cc | 58 … {prim::kPrimMomentum, {2, 3}}, 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/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 60 … prim::kPrimIdentity, prim::kPrimMomentum, prim::kPrimMul, prim::kPrimPow}); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 344 inline const PrimitivePtr kPrimMomentum = std::make_shared<Primitive>("Momentum"); variable
|