Searched refs:backward_params (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 342 void ReorderForParams(const std::vector<AnfNodePtr> &backward_params, const std::vector<AnfNodePtr>… in ReorderForParams() argument
353 if (!backward_params.empty()) { in ReorderForParams()
361 auto post_node2 = backward_params[0]; in ReorderForParams()
582 … std::vector<AnfNodePtr> *forward_params, std::vector<AnfNodePtr> *backward_params, in GetBorderNode() argument
611 backward_params->push_back(node); in GetBorderNode()
635 std::sort((*backward_params).begin(), (*backward_params).end(), CompFunc); in GetBorderNode()
667 std::vector<AnfNodePtr> backward_params; in Reorder() local
669 …de(&forward_start, &forward_end, &backward_start, &backward_end, &forward_params, &backward_params, in Reorder()
683 auto backward_params_pair = Deduplicate(backward_params, root, micro_max); in Reorder()
705 …ReorderForParams(backward_params, forward_params, allreduce_params, forward_params_pair, backward_… in Reorder()
|
| D | pipeline_split_utils.h | 43 void ReorderForParams(const std::vector<AnfNodePtr> &backward_params, const std::vector<AnfNodePtr>…
54 … std::vector<AnfNodePtr> *forward_params, std::vector<AnfNodePtr> *backward_params,
|