Home
last modified time | relevance | path

Searched refs:kReceiveOpName (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_util.cc70 if (op_name == kReceiveOpName) { in GetHcomDataType()
75 if (op_name == kReceiveOpName) { in GetHcomDataType()
134 if (AnfAlgo::GetCNodeName(anf_node) == kReceiveOpName) { in GetHcomCount()
Dhccl_kernel.cc36 {mindspore::kReceiveOpName, mindspore::kHcomOpTypeReceive},
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.h62 …explicit RecvFusion(size_t groups = 1) : CommunicationOpFusion("recv_fusion", kReceiveOpName, grou…
Dcommunication_op_fusion.cc146 if (op_name_ == kHcomSendOpName || op_name_ == kReceiveOpName) { in GetSplitSegments()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h71 constexpr auto kReceiveOpName = "Receive"; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc1196 AnfAlgo::GetCNodeName(kernel) != kReceiveOpName) { in AllocCommunicationOpDynamicRes()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc124 if (node_name == kHcomSendOpName || node_name == kReceiveOpName) { in GetHcomTaskNum()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc1650 … kReduceScatterOpName, kHcomSendOpName, kReceiveOpName, in IsCommunicationOp()