Searched defs:send_node (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_stream_assign.cc | 184 CNodePtr send_node = nullptr; in InsertStreamSwitchNode() local
207 … const CNodePtr &mock_send_node, const CNodePtr &mock_recv_node, CNodePtr *send_node, in GenSendRecvCNodesForAllReduce()
250 const CNodePtr &send_node, const CNodePtr &recv_node) { in CacheSendRecvCNodesForAllReduce()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | process_send_recv_for_ge.cc | 217 void FindFollowing(const AnfNodePtr &send_node, std::map<AnfNodePtr, std::set<AnfNodePtr>> *clouse_… in FindFollowing()
270 for (auto &send_node : all_send) { in AddSendClosureDepend() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_splitter.cc | 172 void SetSendNodeAttr(const AnfNodePtr &send_node, const InterProcessOpEdge &inter_process_edge) { in SetSendNodeAttr()
245 CNodePtr send_node = func_graph->NewCNode(send_inputs); in CreateSendNode() local
1075 for (const auto &send_node : rpc_send_nodes) { in FuseRpcSendNodes() local
1152 CNodePtr send_node = std::get<0>(op_pair); in FuseCommEdges() local
1771 auto send_node = std::get<0>(send_recv_pair); in DumpDistributedGraph() local
1835 auto send_node = CreateSendNode(func_graph_, edge); in GenerateInterProcessOpsForNodeInputs() local
2083 auto send_node = std::get<0>(send_recv_pair); in EliminateExtraNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.cc | 191 void PsEmbeddingCacheInserter::SetSendNodeAttr(const CNodePtr &send_node, int32_t param_key, in SetSendNodeAttr()
310 CNodePtr send_node = graph->NewCNode(send_inputs); in ConstructEmbeddingLookupSubGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | acl_stream_assign.cc | 220 …auto send_node = CreateSendApplyKernel(kernel_graph, event_id, record_stream_id, event_generate_id… in AddBoundarySendRecvKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_interleave.cc | 565 auto send_node = NewValueNode(send_op); in InsertSendReceive() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | graph_scheduler.cc | 2390 auto send_node = pair.first; in LinkControlArrowBySendRecvNodes() local
|