Home
last modified time | relevance | path

Searched defs:node_user_map (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dcame_parallel_handler.h60 const NodeUsersMap &node_user_map; variable
Dcame_parallel_handler.cc96 const NodeUsersMap &node_user_map) { in ForwardSearchCNode()
145 const NodeUsersMap &node_user_map) in CameCommHandler()
Dstep_parallel.cc1503 auto &node_user_map = manager->node_users(); in InsertAllGatherOp() local
3237 auto &node_user_map = manager->node_users(); in InsertAllReduceForNormValue() local
3325 static void InsertDivAndAllReduceForNorm(const NodeUsersMap &node_user_map, const AnfNodePtr &param… in InsertDivAndAllReduceForNorm()
3353 static AnfNodePtr GetMirrorOp(const NodeUsersMap &node_user_map, const AnfNodePtr &parameter) { in GetMirrorOp()
3384 const auto &node_user_map = manager->node_users(); in HandleGlobalNormScale() local
Dstep_auto_parallel.cc1334 auto node_user_map = manager->node_users(); in CalculateMicroBatchSize() local
Dparameter_manager.cc119 auto &node_user_map = node->func_graph()->manager()->node_users(); in FindParameterNodeUsers() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc150 …lidNode(const AnfNodePtr &node, const AnfNodePtr &return_node, const NodeUsersMap &node_user_map) { in IsValidNode()
227 auto node_user_map = manager->node_users(); in SetParameterStartForCellShare() local
385 bool IsSourceUsedByMirror(const CNodePtr &node, const NodeUsersMap &node_user_map) { in IsSourceUsedByMirror()
400 const AnfNodePtr &accu_parameter, const NodeUsersMap &node_user_map) { in InsertVirtualAssignAdd()
887 void BroadCastNeedGrad(const AnfNodePtr &node, NodeUsersMap *node_user_map, const FuncGraphPtr &roo… in BroadCastNeedGrad()
906 auto &node_user_map = manager->node_users(); in LabelNeedGrad() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline_split.cc153 auto node_user_map = manager->node_users(); in InsertVirtualDataset() local
Dpass.cc1001 const auto &node_user_map = manager->node_users(); in RemoveValueNodeDuplicationsPass() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ad/
Ddfunctor.cc922 CNodePtr GetJUser(const NodeUsersMap &node_user_map, const CNodePtr &cnode, int index) { in GetJUser()
1004 const auto &node_user_map = manager->node_users(); in FindPrimalJPair() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dvmap_eliminate.cc433 auto &node_user_map = manager->node_users(); in CopyNodeToVmap() local
473 auto &node_user_map = manager->node_users(); in BindAxis() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dstage_compute.cc268 auto node_user_map = manager->node_users(); in GetNumMicro() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc253 auto node_user_map = manager_->node_users(); in LabelMicroBatch() local
Dpipeline_transformer.cc417 auto node_user_map = manager_->node_users(); in LabelMicroBatch() local