Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_fusion.cc103 VectorRef update_with_lr = VectorRef({prim::kPrimMul, lr_, update}); in DefinePattern() local
Dadam_weight_decay_fusion.cc105 VectorRef update_with_lr = VectorRef({prim::kPrimMul, lr_, new_update}); in DefinePattern() local