Searched refs:accu_parameter (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 112 const AnfNodePtr &accu_parameter) { in InsertVirtualAssignAdd() argument
130 …ePtr> virtual_node_input = {value_node, cnode->input(IntToSize(node_user.second)), accu_parameter}; in InsertVirtualAssignAdd()
177 auto accu_parameter = FindGradAccuParameter(parameters, parameter_ptr->name()); in HandleReceiveParam() local
178 if (!accu_parameter) { in HandleReceiveParam()
192 InsertVirtualAssignAdd(node_user, root->manager(), accu_parameter); in HandleReceiveParam()
195 InsertVirtualAssignAdd(temp_user, root->manager(), accu_parameter); in HandleReceiveParam()
198 InsertVirtualAccuGrad(node, root->manager(), accu_parameter); in HandleReceiveParam()
207 auto accu_parameter = FindGradAccuParameter(parameters, parameter_ptr->name()); in AddVirtualAssignAdd() local
208 if (!accu_parameter) { in AddVirtualAssignAdd()
222 InsertVirtualAssignAdd(node_user, root->manager(), accu_parameter); in AddVirtualAssignAdd()
[all …]
|
| D | pipeline_split_utils.h | 32 const AnfNodePtr &accu_parameter);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 402 … const FuncGraphManagerPtr &manager, const AnfNodePtr &accu_parameter) { in InsertFullySplitParamGradAccu() argument
412 …ePtr> virtual_node_input = {value_node, cnode->input(IntToSize(node_user.second)), accu_parameter}; in InsertFullySplitParamGradAccu()
434 auto accu_parameter = FindGradAccuParameter(parameters, param_ptr->name()); in HandleFullySplitParameters() local
435 if (!accu_parameter) { in HandleFullySplitParameters()
447 InsertFullySplitParamGradAccu(user, root->manager(), accu_parameter); in HandleFullySplitParameters()
|