Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc303 … const PipelinePair &backward_start_pair, const PipelinePair &backward_end_pair, in ReorderForBackward() argument
313 … auto post_node1 = backward_start_pair.first[LongToSize(SizeToLong(i) - stage_num + stage_id + 1)]; in ReorderForBackward()
321 … auto prior_node3 = backward_start_pair.second[LongToSize(SizeToLong(i) - stage_num + stage_id)]; in ReorderForBackward()
329 …for (size_t j = LongToSize(SizeToLong(backward_start_pair.first.size()) - stage_num + stage_id + 1… in ReorderForBackward()
330 j < backward_start_pair.first.size(); ++j) { in ReorderForBackward()
332 auto post_node5 = backward_start_pair.first[j]; in ReorderForBackward()
337 auto post_node6 = backward_start_pair.first[0]; in ReorderForBackward()
640 const PipelinePair &backward_start_pair, const PipelinePair &backward_end_pair, in CheckBorderNode() argument
651 if (backward_start_pair.first.size() != micro_size) { in CheckBorderNode()
652 MS_LOG(EXCEPTION) << "backward_node's size:" << backward_start_pair.first.size() in CheckBorderNode()
[all …]
Dpipeline_split_utils.h41 … const PipelinePair &backward_start_pair, const PipelinePair &backward_end_pair,
69 …const PipelinePair &backward_start_pair, const PipelinePair &backward_end_pair, size_t micro_size);