Searched defs:param_fp32 (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | lamb_fission.cc | 96 … const AnfNodePtr ¶m_fp32, const AnfNodePtr &gradient_fp32, in CreateLambApplyOptimizerAssignNode()
229 auto param_fp32 = CreateCastNode(graph, param_node, kNumberTypeFloat32); in Process() local
|
| D | adam_weight_decay_fission.cc | 98 auto param_fp32 = CreateCastNode(graph, ori_inputs[kIdxParam], kNumberTypeFloat32); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | lamb_fission.cc | 104 … const AnfNodePtr ¶m_fp32, const AnfNodePtr &gradient_fp32, in CreateLambApplyOptimizerAssignNode()
295 auto param_fp32 = CreateCastNode(graph, param_node, kNumberTypeFloat32); in Process() local
|