Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_stream_assign.h44 …return (AnfAlgo::GetCNodeName(cnode) == kRecvOpName && AnfAlgo::GetCNodeName(n.cnode) == kSendOpNa…
Dgpu_stream_assign.cc183 *recv_node = CreateStreamSwitchNode(kernel_graph, kRecvOpName); in GenSendRecvCNodesForAllReduce()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc1885 if (name == kSendOpName || name == kRecvOpName) { in CheckEventAssign()
1919 if (!(first_name == kSendOpName && second_name == kRecvOpName)) { in CheckEventAssign()
1949 auto recv_op = std::make_shared<Primitive>(kRecvOpName); in CreateRecvApplyKernel()
2274 if (name == kSendOpName || name == kRecvOpName) { in GetStreamActiveKind()
2294 … if (AnfAlgo::GetCNodeName(cnode) == kSendOpName || AnfAlgo::GetCNodeName(cnode) == kRecvOpName) { in GetStreamActiveKind()
2398 if (name == kRecvOpName) { in FindEventRelations()
2446 if (next_cnode_name == kSendOpName || next_cnode_name == kRecvOpName) { in AdjustAtomicAddrCleanOrder()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse_checker.cc29 if (node_name == kSendOpName || node_name == kRecvOpName) { in CheckSignalOps()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h169 constexpr auto kRecvOpName = "StreamRecv"; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc96 auto recv_op = std::make_shared<Primitive>(kRecvOpName); in CreateRecvApplyKernel()