Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dadam_fusion.cc140 auto gradient_input = utils::cast<AnfNodePtr>((*equiv)[gradient_]); in Process() local
Dadam_weight_decay_fusion.cc144 auto gradient_input = utils::cast<AnfNodePtr>((*equiv)[gradient_]); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc841 …tr, CNodePtr> ParameterServerMode::CreateNodesForGradAccumulation(const AnfNodePtr &gradient_input, in CreateNodesForGradAccumulation()