Home
last modified time | relevance | path

Searched refs:PIPELINE_PARAM (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc172 if (!cnode->HasPrimalAttr(PIPELINE_PARAM)) { in HandleReceiveParam()
610 if (cnode->HasPrimalAttr(PIPELINE_PARAM)) { in GetBorderNode()
626 if (cnode->HasPrimalAttr(PIPELINE_PARAM)) { in GetBorderNode()
724 if (cnode->HasPrimalAttr(PIPELINE_PARAM)) { in ReorderForPredict()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc601 send->AddPrimalAttr(PIPELINE_PARAM, value); in InsertSend()
664 recv->set_user_data<AnfNode>(PIPELINE_PARAM, node); in InsertReceive()
665 recv->AddPrimalAttr(PIPELINE_PARAM, value); in InsertReceive()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h409 constexpr char PIPELINE_PARAM[] = "pipeline_param"; variable
Doperator_info.cc347 param = param_node->user_data<AnfNode>(PIPELINE_PARAM)->cast<ParameterPtr>(); in AddCommOpFusionType()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2574 if (!cnode->HasPrimalAttr(PIPELINE_PARAM)) { in ParallelCommunication()