Home
last modified time | relevance | path

Searched defs:load_param (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dauto_monad_eliminate.cc66 auto load_param = cnode->input(1); in GenerateLoadGroups() local
296 auto load_param = toposet[group.back()]->cast<CNodePtr>()->input(1); in ReplaceAutoMonadNode() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_fusion.cc91 VectorRef load_param = VectorRef({prim::kPrimLoad, param_, u_}); in DefinePattern() local
Dadam_weight_decay_fusion.cc91 VectorRef load_param = VectorRef({prim::kPrimLoad, param_, u_}); in DefinePattern() local