Searched defs:node_stage (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | fold_pipeline_transformer.cc | 223 …neTransformer::InsertSend(const AnfNodePtr ¶meter, int64_t user_node_stage, int64_t node_stage, in InsertSend()
305 int64_t FoldPipelineTransformer::ComputeRecvTag(int64_t node_stage, int64_t user_node_stage, int64_… in ComputeRecvTag()
330 int64_t node_stage, const ValuePtr &value, in InsertReceive()
487 bool IsStageConflict(int64_t node_stage, int64_t user_node_stage, int64_t node_segment, int64_t use… in IsStageConflict()
506 auto node_stage = stage_info->stage(); in CutBorderForNode() local
|
| D | pipeline_transformer.cc | 1000 …neTransformer::InsertSend(const AnfNodePtr ¶meter, int64_t user_node_stage, int64_t node_stage, in InsertSend()
1051 int64_t node_stage, const ValuePtr &value, in InsertReceive()
1228 auto node_stage = stage_info->stage(); in CutBorderForNode() local
|
| D | pipeline_interleave.cc | 553 auto node_stage = node_stage_info->stage(); in InsertSendReceive() local
608 auto node_stage = stage_info->stage(); in CutBorderForNode() local
|