Searched refs:AnfNodeIndexSet (Results 1 – 17 of 17) sorted by relevance
| /third_party/mindspore/mindspore/core/api/ir/ |
| D | func_graph_manager.h | 37 using AnfNodeIndexSet = CompactSet<std::pair<AnfNodePtr, int>>; variable
38 using NodeUsersMap = std::unordered_map<AnfNodePtr, AnfNodeIndexSet, PointerHash<AnfNodePtr>>;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | adjust_depend_for_parallel_optimizer_recompute_all_gather.cc | 108 AnfNodeIndexSet allgather_node_set = manager->node_users()[cnode]; in AdjustAllgatherDepend()
128 AnfNodeIndexSet allgather_node_set = manager->node_users()[cnode]; in AdjustAllgatherDepend()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.h | 34 using ParameterUsersInfo = std::pair<std::string, std::pair<AnfNodePtr, AnfNodeIndexSet>>;
|
| D | step_parallel.cc | 548 AnfNodeIndexSet node_set = manager->node_users()[node]; in StepRedistribution()
691 AnfNodeIndexSet node_set = manager->node_users()[node]; in StepSplitTensor()
1003 AnfNodeIndexSet node_set = manager->node_users()[anode]; in FindCNode()
1424 AnfNodeIndexSet node_set = manager->node_users()[node]; in FindParallelCareNode()
1460 AnfNodeIndexSet param_sub_set = manager->node_users()[parameter]; in FindSubGraph()
1971 AnfNodeIndexSet node_set = manager->node_users()[cnode]; in FindNextLayout()
2046 AnfNodeIndexSet node_set = manager->node_users()[node]; in FindParameterNextLayout()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.h | 67 …AnfNodeIndexSet GetActualOpUsers(const std::pair<AnfNodePtr, int> &node_pair, NodeUsersMap *node_u…
|
| D | pipeline_transformer.cc | 362 AnfNodeIndexSet PipelineTransformer::GetActualOpUsers(const std::pair<AnfNodePtr, int> &node_pair, in GetActualOpUsers()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | manager.h | 58 using AnfNodeIndexSet = api::AnfNodeIndexSet; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | confusion_mul_grad_fusion.cc | 66 const AnfNodeIndexSet &outputs_set = manager->node_users()[input2]; in GetMul0()
|
| D | lamb_next_mv_with_decay_v1_rule.cc | 161 AnfNodeIndexSet mul4_output_node_index_set = manager->node_users()[mul4]; in Process()
|
| D | lamb_next_mv_rule.cc | 40 AnfNodeIndexSet real_div0_outputs = users[real_div0]; in IsRuleMatched()
|
| D | lamb_next_mv_with_decay_rule.cc | 102 AnfNodeIndexSet mul4_outputs = manager->node_users()[mul4]; in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_swap_manager.cc | 189 AnfNodeIndexSet node_set = iter->second; in IsCommunicationRelevantOp()
234 AnfNodeIndexSet node_set = iter->second; in SaveUserKernelTopoOrder()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | decrease_transfer_precision.cc | 80 bool IsAllUserCandidateNode(const AnfNodeIndexSet &users) { in IsAllUserCandidateNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_fusion.cc | 186 AnfNodeIndexSet node_set = manager->node_users()[para]; in FindMirror()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 371 AnfNodeIndexSet node_set = manager->node_users()[cnode]; in FindReshapeNextNodeStraCosts()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 305 …static AnfNodeIndexSet GetUpdateStateUsers(const FuncGraphManagerPtr &manager, const AnfNodePtr &n…
|
| D | anf_runtime_algorithm.cc | 2285 AnfNodeIndexSet AnfRuntimeAlgorithm::GetUpdateStateUsers(const FuncGraphManagerPtr &manager, const … in GetUpdateStateUsers()
2286 AnfNodeIndexSet update_states; in GetUpdateStateUsers()
|