Home
last modified time | relevance | path

Searched refs:PIPELINE_END (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc496 if (prim && prim->HasAttr(PIPELINE_END)) { in LastStageEndNode()
502 if (!temp_prim || temp_prim->HasAttr(PIPELINE_END)) { in LastStageEndNode()
516 end_cnode->AddPrimalAttr(PIPELINE_END, end_cnode->GetPrimalAttr(MICRO)); in LastStageEndNode()
601 if (cnode->HasPrimalAttr(PIPELINE_END)) { in GetBorderNode()
623 if (cnode->HasPrimalAttr(PIPELINE_END)) { in GetBorderNode()
721 if (cnode->HasPrimalAttr(PIPELINE_END)) { in ReorderForPredict()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h410 constexpr char PIPELINE_END[] = "pipeline_end"; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc599 send->AddPrimalAttr(PIPELINE_END, value); in InsertSend()