Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dadam_fusion.cc108 VectorRef update_with_lr = VectorRef({prim::kPrimMul, lr_, update}); in DefinePattern() local
Dadam_weight_decay_fusion.cc110 VectorRef update_with_lr = VectorRef({prim::kPrimMul, lr_, new_update}); in DefinePattern() local