Searched defs:recv_node (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_stream_assign.cc | 185 CNodePtr recv_node = nullptr; in InsertStreamSwitchNode() local
208 CNodePtr *recv_node) { in GenSendRecvCNodesForAllReduce()
250 const CNodePtr &send_node, const CNodePtr &recv_node) { in CacheSendRecvCNodesForAllReduce()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.cc | 219 void PsEmbeddingCacheInserter::SetRecvNodeAttr(const CNodePtr &recv_node, const std::string &src_ro… in SetRecvNodeAttr() argument
461 CNodePtr recv_node = root_graph_->NewCNode(recv_inputs); in CreateRecvNode() local
535 CNodePtr recv_node = CreateRecvNode(); in ConstructEmbeddingCacheGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | overlap_opt_shard_in_pipeline.cc | 54 auto recv_node = node->cast<CNodePtr>(); in is_first_receive() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_splitter.cc | 197 void SetRecvNodeAttr(const AnfNodePtr &recv_node, const InterProcessOpEdge &inter_process_edge) { in SetRecvNodeAttr()
260 CNodePtr recv_node = nullptr; in CreateRecvNode() local
1108 for (const auto &recv_node : rpc_recv_nodes) { in FuseRpcRecvNodes() local
1772 auto recv_node = std::get<1>(send_recv_pair); in DumpDistributedGraph() local
1840 auto recv_node = CreateRecvNode(func_graph_, edge); in GenerateInterProcessOpsForNodeInputs() local
2084 auto recv_node = std::get<1>(send_recv_pair); in EliminateExtraNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | process_send_recv_for_ge.cc | 181 for (auto &recv_node : recv_nodes) { in AddAllGatherRecvDepend() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | acl_stream_assign.cc | 222 …auto recv_node = CreateRecvApplyKernel(kernel_graph, event_id, record_stream_id, wait_stream_id, e… in AddBoundarySendRecvKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 1206 …auto recv_node = InsertReceive(graph, argument, use_node, SizeToInt(pos), user_stage, stage, micro… in HandleParameterGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | graph_scheduler.cc | 2391 auto recv_node = pair.second; in LinkControlArrowBySendRecvNodes() local
|