Searched refs:allreduce_params (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.h | 44 … const std::vector<AnfNodePtr> &allreduce_params, const PipelinePair &forward_params_pair,
55 std::vector<AnfNodePtr> *allreduce_params, const FuncGraphPtr &root);
|
| D | pipeline_split_utils.cc | 343 … const std::vector<AnfNodePtr> &allreduce_params, const PipelinePair &forward_params_pair, in ReorderForParams() argument
354 if (!allreduce_params.empty()) { in ReorderForParams()
355 for (auto &node : allreduce_params) { in ReorderForParams()
583 std::vector<AnfNodePtr> *allreduce_params, const FuncGraphPtr &root) { in GetBorderNode() argument
614 allreduce_params->push_back(node); in GetBorderNode()
668 std::vector<AnfNodePtr> allreduce_params; in Reorder() local
670 &allreduce_params, root); in Reorder()
705 …ReorderForParams(backward_params, forward_params, allreduce_params, forward_params_pair, backward_… in Reorder()
|