Home
last modified time | relevance | path

Searched defs:node_users_map (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Deliminate_maketuple_getitem.cc28 const auto &node_users_map = manager->node_users(); in Run() local
Dinsert_depend_for_grad_comm.cc96 auto node_users_map = manager->node_users(); in GetOutputNodesSkipComm() local
222 CNodePtr RelyNode(const NodeUsersMap &node_users_map, const std::set<AnfNodePtr, AnfNodeCmp> &input… in RelyNode()
301 auto node_users_map = manager->node_users(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Ddropout_gen_mask_fusion.cc30 auto &node_users_map = manager->node_users(); in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc361 CNodePtr FindNodeWithMircoSize(const AnfNodePtr &node_user, const NodeUsersMap &node_users_map) { in FindNodeWithMircoSize()
486 const std::pair<AnfNodePtr, int> &node_ptr, const NodeUsersMap &node_users_map, in FindNextNode()
517 auto node_users_map = parameter->func_graph()->manager()->node_users(); in FuncNodeUsersSet() local
541 auto node_users_map = root->manager()->node_users(); in HandleReceiveParam() local
589 auto node_users_map = root->manager()->node_users(); in AddVirtualAssignAdd() local
872 void BroadCastMicroBatch(const CNodePtr &node, NodeUsersMap *node_users_map, const ValuePtr &value,… in BroadCastMicroBatch()
951 ValuePtr Micro(const CNodePtr &cnode, NodeUsersMap *node_users_map, size_t max_depth) { in Micro()
970 auto &node_users_map = manager->node_users(); in ParameterStartNode() local
996 auto &node_users_map = manager->node_users(); in HandleMicroBatch() local
Dgrad_accumulation_utils.cc44 auto node_users_map = manager->node_users(); in TagMicroBatchStart() local
110 auto node_users_map = manager->node_users(); in TagMicroBatchBpEndInCellShare() local
Dgraph_utils.cc1083 auto node_users_map = manager->node_users(); in UpdateTupleGetItemShapeValue() local
1218 auto node_users_map = func_graph->manager()->node_users(); in UpdateShapeNode() local
Dfold_pipeline_split_utils.cc140 auto node_users_map = root->manager()->node_users(); in FindNodeFirstUser() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc433 const NodeUsersMap &node_users_map, in GetStageByArgument()
473 auto &node_users_map = manager_->node_users(); in ParameterColoring() local
506 auto node_users_map = manager_->node_users(); in RemoveMonadNode() local
696 auto node_users_map = manager_->node_users(); in ElimParameter() local
1041 auto node_users_map = manager_->node_users(); in PartitionChunkGraph() local
1127 auto node_users_map = manager_->node_users(); in HandleSendParam() local
1200 auto node_users_map = manager_->node_users(); in FreezeGradient() local
1244 static AnfNodePtr GetDout(const AnfNodePtr &node, const NodeUsersMap &node_users_map) { in GetDout()
1295 const auto &node_users_map = manager->node_users(); in IsolatedNodeAttach() local
Dpipeline_transformer.cc679 AnfNodeIndexSet GetActualOpUsers(const AnfNodePtr &node, NodeUsersMap *node_users_map) { in GetActualOpUsers()
715 auto node_users_map = manager_->node_users(); in GetParameterPair() local
786 const auto &node_users_map = manager_->node_users(); in HandleSharedParameter() local
848 const NodeUsersMap &node_users_map, in GetStageByArgument()
885 auto &node_users_map = manager_->node_users(); in ParameterColoring() local
917 auto node_users_map = manager_->node_users(); in RemoveMonadNode() local
1770 auto &node_users_map = manager_->node_users(); in HandleGraphInputs() local
1932 auto node_users_map = manager_->node_users(); in FreezeGradient() local
1981 auto node_users_map = manager_->node_users(); in ElimParameter() local
Dfold_pipeline_transformer.cc611 const auto &node_users_map = manager_->node_users(); in HandleSharedParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dincorporate_call.h135 auto &node_users_map = manager->node_users(); in MakeNewNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/
Dmem_swap_manager.cc131 auto &node_users_map = kernel_exec_info.node_users_map_; in CheckDistanceBetweenKernels() local
155 auto &node_users_map = kernel_exec_info.node_users_map_; in CheckDistanceBetweenKernelsWithIdx() local
712 const auto &node_users_map = kernel_exec_info.node_users_map_; in DumpUserNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline_split.cc109 auto node_users_map = root->manager()->node_users(); in FindForwardGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dstorage_format_convertor.cc88 auto node_users_map = manager->node_users(); in GetOutputNodesSkipVirtualNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc455 const AnfNodePtr &node, const FuncGraphManagerPtr &manager, const NodeUsersMap &node_users_map, in RedistributionNextNode()
854 …(std::queue<std::pair<std::shared_ptr<AnfNode>, int>> *visited, const NodeUsersMap &node_users_map, in AddVisitedNode()
872 … const NodeUsersMap &node_users_map, const int index, in BFSParallelCareNode()
920 AnfNodePtr GetChildCastNode(const AnfNodePtr &node_ptr, const NodeUsersMap &node_users_map) { in GetChildCastNode()
964 …hildCastWithFP32ToFP16(const std::pair<AnfNodePtr, int> &res, const NodeUsersMap &node_users_map) { in FindChildCastWithFP32ToFP16()
Dstep_parallel.cc581 static void StepRedistribution(const CNodePtr &cnode, const NodeUsersMap &node_users_map) { in StepRedistribution()
1702 const auto &node_users_map = manager->node_users(); in CoverSliceShape() local
2730 const auto &node_users_map = manager->node_users(); in ParallelCommunication() local
3284 …NodePtr FindExpandDimsWIthGradScale(const AnfNodePtr &node_ptr, const NodeUsersMap &node_users_map, in FindExpandDimsWIthGradScale()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doverlap_recompute_and_grad_model_parallel.cc390 auto node_users_map = manager->node_users(); in ReorderRecomputeAndGradNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dgrouped_pairwise_exchange_alltoall.cc86 auto node_users_map = manager->node_users(); in FindBackAlltoall() local