Home
last modified time | relevance | path

Searched refs:forward_end_before_pair (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc304 const PipelinePair &forward_end_before_pair, const FuncGraphPtr &root) { in ReorderForBackward() argument
312 auto prior_node1 = forward_end_before_pair.second[i]; in ReorderForBackward()
336 auto prior_node6 = forward_end_before_pair.second[LongToSize(stage_num - 1 - stage_id)]; in ReorderForBackward()
685 PipelinePair forward_end_before_pair; in Reorder() local
691 forward_end_before_pair.first.push_back(temp_node); in Reorder()
697 forward_end_before_pair.second.push_back(temp_node); in Reorder()
700 forward_end_before_pair = forward_end_pair; in Reorder()
704 forward_end_before_pair, root); in Reorder()
Dpipeline_split_utils.h42 const PipelinePair &forward_end_before_pair, const FuncGraphPtr &root);