Home
last modified time | relevance | path

Searched refs:has_user_data (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc76 if ((prim->name() == CAST) && !cnode->has_user_data<OperatorInfo>()) { in IsParallelCareNode()
Dstep_parallel.cc581 if (IsParallelCareNode(use_cnode) && use_cnode->has_user_data<OperatorInfo>()) { in StepRedistribution()
1439 if (IsParallelCareNode(cnode) && cnode->has_user_data<OperatorInfo>()) { in FindParallelCareNode()
1989 if (IsParallelCareNode(use_apply) && use_apply->has_user_data<OperatorInfo>()) { in FindNextLayout()
1996 << " " << use_apply->has_user_data<OperatorInfo>(); in FindNextLayout()
2028 if (IsParallelCareNode(cnode) && cnode->has_user_data<OperatorInfo>()) { in FindPrevParallelCareNodeLayout()
2066 if (IsParallelCareNode(use_apply) && use_apply->has_user_data<OperatorInfo>()) { in FindParameterNextLayout()
2150 if (IsParallelCareNode(cnode) && cnode->has_user_data<OperatorInfo>() && in FindPrevLayout()
2191 if (!IsParallelCareNode(cnode) || !cnode->has_user_data<OperatorInfo>()) { in ReshapeInit()
2236 (IsPrimitiveCNode(cnode, prim::kPrimCast) && !cnode->has_user_data<OperatorInfo>())) { in HandleDependLoss()
2566 if (!IsParallelCareNode(cnode) || !cnode->has_user_data<OperatorInfo>()) { in ParallelCommunication()
[all …]
Dparameter_manager.cc114 …if (cnode == nullptr || !cnode->has_user_data<OperatorInfo>() || IsSomePrimitive(cnode, RECEIVE)) { in FindParameterNodeUsers()
121 if (c == nullptr || !c->has_user_data<OperatorInfo>() || IsSomePrimitive(c, RECEIVE)) { in FindParameterNodeUsers()
/third_party/mindspore/mindspore/core/ir/
Danf.h194 bool has_user_data(const std::string &key) const { return user_data_.has(key); } in has_user_data() function
197 bool has_user_data() const { in has_user_data() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_fusion.cc53 if (IsParallelCareNode(cnode) && cnode->has_user_data<OperatorInfo>()) { in FindCNodesWithPara()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc285 if (!IsParallelCareNode(cnode) || !cnode->has_user_data<OperatorInfo>()) { in FindReshape()