Searched refs:node_stage (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 555 …neTransformer::InsertSend(const AnfNodePtr ¶meter, int64_t user_node_stage, int64_t node_stage, in InsertSend() argument
557 auto dest_rank = global_rank_ + (user_node_stage - node_stage) * per_stage_rank_num_; in InsertSend()
622 int64_t node_stage, const ValuePtr &value, in InsertReceive() argument
624 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive()
634 Attr attr_rank = std::make_pair(SRC_RANK, MakeValue(node_stage)); in InsertReceive()
833 auto node_stage = node->stage(); in CutBorder() local
835 if (node_stage != stage_ && user_node_stage != stage_) { in CutBorder()
843 if (node_stage < user_node_stage) { in CutBorder()
844 if (node_stage == stage_) { in CutBorder()
846 … auto send_depend = HandleParameterGraph(node, user_node, node_stage, user_node_stage, micro, in CutBorder()
[all …]
|
| D | pipeline_transformer.h | 72 …SendAttr InsertSend(const AnfNodePtr ¶meter, int64_t user_node_stage, int64_t node_stage, cons…
74 int64_t user_node_stage, int64_t node_stage, const ValuePtr &value,
|