Home
last modified time | relevance | path

Searched defs:gradient_fp32 (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dlamb_fission.cc96 … const AnfNodePtr &param_fp32, const AnfNodePtr &gradient_fp32, in CreateLambApplyOptimizerAssignNode()
231 auto gradient_fp32 = CreateCastNode(graph, ori_inputs[kGradientIndex], kNumberTypeFloat32); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dlamb_fission.cc104 … const AnfNodePtr &param_fp32, const AnfNodePtr &gradient_fp32, in CreateLambApplyOptimizerAssignNode()
297 auto gradient_fp32 = CreateCastNode(graph, ori_inputs[kGradientIndex], kNumberTypeFloat32); in Process() local