Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc238 auto micro1 = cnode1->GetPrimalAttr(MICRO); in CompFunc()
239 auto micro2 = cnode2->GetPrimalAttr(MICRO); in CompFunc()
370 auto micro_value = cnode->GetPrimalAttr(MICRO); in GetMicroBatch()
516 end_cnode->AddPrimalAttr(PIPELINE_END, end_cnode->GetPrimalAttr(MICRO)); in LastStageEndNode()
527 return cnode->GetPrimalAttr(MICRO); in Micro()
566 auto micro = cnode->GetPrimalAttr(MICRO); in HandleMicroBatch()
596 auto forward_node_name = cnode->GetPrimalAttr(kPrimalAttrForwardNodeName); in GetBorderNode()
674 auto micro_size = forward_end_cnode->GetPrimalAttr(MICRO); in Reorder()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc429 auto micro = cnode->GetPrimalAttr(MICRO); in HandleSharedParameter()
838 auto micro = user_node->cast<CNodePtr>()->GetPrimalAttr(MICRO); in CutBorder()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc225 auto param_index = IntToSize(GetValue<int>(user_cnode->GetPrimalAttr(PARAM_INDEX))); in GetParameterSliceInfo()
Dstep_parallel.cc1140 auto param_index = GetValue<int>(node->GetPrimalAttr(PARAM_INDEX)); in InsertMirrorOps()
1934 strategyPtr = ExtractStrategy(cnode->GetPrimalAttr(STRATEGY)); in ExtractInformation()
/third_party/mindspore/mindspore/core/ir/
Danf.h320 ValuePtr GetPrimalAttr(const std::string &name) const { in GetPrimalAttr() function