Searched refs:AddPrimalAttr (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 86 cnode->AddPrimalAttr(MICRO, MakeValue(micro)); in SetMicroBatch()
87 cnode->AddPrimalAttr(PIPELINE_BEGIN, MakeValue(micro)); in SetMicroBatch()
599 send->AddPrimalAttr(PIPELINE_END, value); in InsertSend()
601 send->AddPrimalAttr(PIPELINE_PARAM, value); in InsertSend()
603 send->AddPrimalAttr(PARAM_INDEX, MakeValue(index)); in InsertSend()
605 send->AddPrimalAttr(MICRO, value); in InsertSend()
665 recv->AddPrimalAttr(PIPELINE_PARAM, value); in InsertReceive()
667 recv->AddPrimalAttr(PIPELINE_BEGIN, value); in InsertReceive()
669 recv->AddPrimalAttr(MICRO, value); in InsertReceive()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 108 cnode->AddPrimalAttr(STRATEGY, strategy); in SetStridedSliceStrategy()
423 user_node->AddPrimalAttr(MICRO, value); in BroadCastMicroBatch()
439 cnode->AddPrimalAttr(NEED_GRAD, MakeValue(1)); in BroadCastNeedGrad()
516 end_cnode->AddPrimalAttr(PIPELINE_END, end_cnode->GetPrimalAttr(MICRO)); in LastStageEndNode()
550 cnode->AddPrimalAttr(MICRO, micro); in ParameterStartNode()
551 cnode->AddPrimalAttr(PARAMETER_START, micro); in ParameterStartNode()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.h | 318 void AddPrimalAttr(const std::string &name, const ValuePtr &attr) { primal_attrs_[name] = attr; } in AddPrimalAttr() function
|