Home
last modified time | relevance | path

Searched defs:grad_node (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Doptimize_gradients_allreduce_overlap.cc361 for (auto &grad_node : grads_list) { in OverLapGradientsAllReduceAndCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_grad.cc719 auto grad_node = GetGradNodeByIndex(weight); in GetWeightGrad() local
Dir_pass.cc511 AnfNodePtr IrPassForward::PassBackwardHook(const ValuePtr &value, const AnfNodePtr &grad_node) { in PassBackwardHook()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc1333 auto grad_node = GetDout(get_item_usr, node_users_map); in IsolatedNodeAttach() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dgraph_build.cc3307 static bool GetGradSens(ValueNode *grad_node) { in GetGradSens()
3341 ValueNode *grad_node = grad_func_node->input(0); in SetGradFuncInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc3013 auto grad_node = FindGrad(cnode, 0); in HandleRootReshapeAndSaveStrategy() local