Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_stream_assign.h44 …fAlgo::GetCNodeName(cnode) == kRecvOpName && AnfAlgo::GetCNodeName(n.cnode) == kSendOpName) ? false
Dgpu_stream_assign.cc181 *send_node = CreateStreamSwitchNode(kernel_graph, kSendOpName); 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()
1929 auto send_op = std::make_shared<Primitive>(kSendOpName); in CreateSendApplyKernel()
2274 if (name == kSendOpName || name == kRecvOpName) { in GetStreamActiveKind()
2294 … if (AnfAlgo::GetCNodeName(cnode) == kSendOpName || AnfAlgo::GetCNodeName(cnode) == kRecvOpName) { in GetStreamActiveKind()
2394 if (name == kSendOpName) { 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.h168 constexpr auto kSendOpName = "StreamSend"; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc76 auto send_op = std::make_shared<Primitive>(kSendOpName); in CreateSendApplyKernel()