Searched defs:param_node_pair (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_fusion.cc | 101 std::pair<AnfNodePtr, bool> param_node_pair = FindParameter(cnode->input(1), func_graph); in SetFusionBySize() local
117 …std::pair<AnfNodePtr, bool> param_node_pair = FindParameterWithAllgather(cnode->input(1), func_gra… in SetFusionBySize() local
153 std::pair<AnfNodePtr, bool> param_node_pair = in SetFusionBySizeReduceScatter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | grad_accumulation_utils.cc | 176 … std::pair<AnfNodePtr, bool> param_node_pair = FindParameter(cnode->input(i), parallel_care_graph); in TagMicroBatchBpEndPrim() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 1373 auto param_node_pair = FindParameter(mirror_cnode->input(1), node->func_graph()); in HandleMirrorInAdaSum() local
|
| D | step_parallel.cc | 1225 std::pair<AnfNodePtr, bool> param_node_pair = FindParameter(node->input(index), func_graph); in DoInsertMirrorOps() local
|