Searched refs:IsSomePrimitive (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.h | 28 bool IsSomePrimitive(const CNodePtr &cnode, const std::string &name);
|
| D | parameter_manager.cc | 114 …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()
186 if (!IsSomePrimitive(cnode, J) || !IsValueNode<FuncGraph>(cnode->input(1))) { in IsUsedParameter()
|
| D | step_parallel_utils.cc | 48 bool IsSomePrimitive(const CNodePtr &cnode, const std::string &name) { in IsSomePrimitive() function
|
| D | step_parallel.cc | 838 if (IsSomePrimitive(node, DROPOUT_DO_MASK)) { in InsertVirtualDivOp()
988 while (IsInTrivialNodeList(node) || IsSomePrimitive(node, LOAD)) { in SkipTrivialNodesMoveDown()
1088 while (!IsSomePrimitive(node, LOAD)) { in SkipTrivialNodesMoveUp()
1095 if (IsSomePrimitive(prev_node, DEPEND)) { in SkipTrivialNodesMoveUp()
1097 if (IsSomePrimitive(node, LOAD)) { in SkipTrivialNodesMoveUp()
1495 if (IsSomePrimitive(res, LOAD)) { in InsertAllGatherAfterCast()
1499 if (!IsSomePrimitive(res, CAST)) { in InsertAllGatherAfterCast()
2506 if (!IsSomePrimitive(expect_tuple_getitem_cnode, prim::kTupleGetItem)) { in GetSensLossPairs()
2525 if (!IsSomePrimitive(expect_j_cnode, J)) { in GetSensLossPairs()
2596 …if (!IsParallelCareNode(cnode) || !cnode->has_user_data<OperatorInfo>() || IsSomePrimitive(cnode, … in ParallelCommunication()
[all …]
|