Searched refs:forward_start_pair (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 302 void ReorderForBackward(const PipelinePair &forward_start_pair, const PipelinePair &forward_end_pai… in ReorderForBackward() argument
311 for (size_t i = LongToSize(stage_num - stage_id); i < (forward_start_pair.first.size()); ++i) { in ReorderForBackward()
316 auto post_node2 = forward_start_pair.first[i]; in ReorderForBackward()
319 …for (size_t i = LongToSize(stage_num - stage_id); i < (forward_start_pair.first.size() + 1); ++i) { in ReorderForBackward()
345 const PipelinePair &forward_start_pair, const FuncGraphPtr &root) { in ReorderForParams() argument
350 auto post_node = forward_start_pair.first[0]; in ReorderForParams()
639 void CheckBorderNode(const PipelinePair &forward_start_pair, const PipelinePair &forward_end_pair, in CheckBorderNode() argument
643 if (forward_start_pair.first.size() != micro_size) { in CheckBorderNode()
644 MS_LOG(EXCEPTION) << "forward_node's size:" << forward_start_pair.first.size() in CheckBorderNode()
680 auto forward_start_pair = Deduplicate(forward_start, root, micro_max); in Reorder() local
[all …]
|
| D | pipeline_split_utils.h | 40 void ReorderForBackward(const PipelinePair &forward_start_pair, const PipelinePair &forward_end_pai…
46 const PipelinePair &forward_start_pair, const FuncGraphPtr &root);
68 void CheckBorderNode(const PipelinePair &forward_start_pair, const PipelinePair &forward_end_pair,
|