Searched refs:NodeUsersMap (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/core/api/ir/ |
| D | func_graph_manager.h | 38 using NodeUsersMap = std::unordered_map<AnfNodePtr, AnfNodeIndexSet, PointerHash<AnfNodePtr>>; variable
48 virtual const NodeUsersMap &node_users() const = 0;
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.h | 59 void BroadCastMicroBatch(const CNodePtr &node, NodeUsersMap *node_users_map, const ValuePtr &value,…
61 void BroadCastNeedGrad(const AnfNodePtr &node, NodeUsersMap *node_user_map, const FuncGraphPtr &roo…
67 ValuePtr Micro(const CNodePtr &cnode, NodeUsersMap *node_users_map, size_t max_depth);
|
| D | pipeline_split_utils.cc | 413 void BroadCastMicroBatch(const CNodePtr &node, NodeUsersMap *node_users_map, const ValuePtr &value,… in BroadCastMicroBatch()
428 void BroadCastNeedGrad(const AnfNodePtr &node, NodeUsersMap *node_user_map, const FuncGraphPtr &roo… in BroadCastNeedGrad()
522 ValuePtr Micro(const CNodePtr &cnode, NodeUsersMap *node_users_map, size_t max_depth) { in Micro()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | manager.h | 60 using NodeUsersMap = api::NodeUsersMap; variable
316 NodeUsersMap &node_users() { return node_users_; } in node_users()
318 const NodeUsersMap &node_users() const final { return node_users_; } in node_users()
340 NodeUsersMap node_users_;
|
| D | manager.cc | 218 node_users_ = NodeUsersMap(); in Reset()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | prim_bprop_optimizer.h | 114 void AnalysisNodeUsingInfo(const NodeUsersMap &node_users, const std::shared_ptr<AnfNode> ¶m,
119 void AalysisForTupleGetItem(const NodeUsersMap &node_users, const std::shared_ptr<AnfNode> ¶m,
|
| D | prim_bprop_optimizer.cc | 75 void PrimBpropOptGraphLevel2Info::AnalysisNodeUsingInfo(const NodeUsersMap &node_users, in AnalysisNodeUsingInfo()
106 void PrimBpropOptGraphLevel2Info::AalysisForTupleGetItem(const NodeUsersMap &node_users, in AalysisForTupleGetItem()
|
| D | dfunctor.cc | 759 CNodePtr GetJUser(const NodeUsersMap &node_user_map, const CNodePtr &cnode, int index) { in GetJUser()
|
| /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 | 363 NodeUsersMap *node_users_map) { in GetActualOpUsers()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_swap_manager.cc | 185 NodeUsersMap &user_map = graph_manager->node_users(); in IsCommunicationRelevantOp()
206 NodeUsersMap &user_map = graph_manager->node_users(); in IsInplaceRelevantOp()
228 NodeUsersMap &user_map = graph_manager->node_users(); in SaveUserKernelTopoOrder()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | segment_runner.cc | 45 AnfNodePtrList GetOutput(const AnfNodePtrList &nodes, const NodeUsersMap &users, in GetOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_converter.cc | 183 const NodeUsersMap &users = manager->node_users(); in RemoveParameterWithoutUser()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | infer_mindir.cc | 70 NodeUsersMap nodeuser_map_;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 592 const NodeUsersMap &users = mng->node_users(); in GetInputIndex()
|