Searched refs:kPrimReceive (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_tensor_move_for_hccl_op.cc | 75 if (IsPrimitiveCNode(cur_node, prim::kPrimReceive)) { in NeedInsertTensorMove()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 156 } else if (IsPrimitiveCNode(cnode, prim::kPrimReceive)) { in GetGeNodeName()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.cc | 235 if (AnfAlgo::GetCNodeName(prior_op) == prim::kPrimReceive->name() || in MergeCastToPriorOp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 219 …= MakeSubstitution(std::make_shared<ReceiveEliminater>(), "receive_eliminate", prim::kPrimReceive); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 114 if (IsPrimitiveCNode(cnode, prim::kPrimReceive) || !cnode->in_forward_flag()) { in InsertVirtualAssignAdd()
168 if (!IsPrimitiveCNode(node, prim::kPrimReceive)) { in HandleReceiveParam()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 465 if (IsPrimitiveCNode(middle_node, prim::kPrimReceive)) { in Redistribution()
1226 if (IsPrimitiveCNode(node, prim::kPrimReceive)) { in BackwardCommunication()
1435 …_prim->name() == DEPEND && node_pair.second != 1) || IsPrimitiveCNode(cnode, prim::kPrimReceive) || in FindParallelCareNode()
1578 if (cnode->in_forward_flag() && !IsPrimitiveCNode(cnode, prim::kPrimReceive) && in ApplyParallelOptOnParam()
2147 if (IsPrimitiveCNode(node, prim::kPrimReceive)) { in FindPrevLayout()
2416 if (IsPrimitiveCNode(cnode, prim::kPrimReceive)) { in InsertForwardOps()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 127 if (!IsPrimitiveCNode(node, prim::kPrimReceive) || node->func_graph() == nullptr) { in operator()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 346 if (IsPrimitiveCNode(node, prim::kPrimReceive)) { in GetOpInfo()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 419 inline const PrimitivePtr kPrimReceive = std::make_shared<Primitive>("Receive"); variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 346 if (IsPrimitiveCNode(param_node, prim::kPrimReceive)) { in AddCommOpFusionType()
|