Home
last modified time | relevance | path

Searched refs:recv_node_ptr (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc101 CNodePtr recv_node_ptr = graph_ptr->NewCNode(recv_input_list); in CreateRecvApplyKernel() local
102 MS_EXCEPTION_IF_NULL(recv_node_ptr); in CreateRecvApplyKernel()
105 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_builder.Build(), recv_node_ptr.get()); in CreateRecvApplyKernel()
106 AnfAlgo::SetNodeAttr(kAttrEventId, MakeValue(event_id), recv_node_ptr); in CreateRecvApplyKernel()
109 recv_node_ptr->set_abstract(abstract_none); in CreateRecvApplyKernel()
110 return recv_node_ptr; in CreateRecvApplyKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc1954 CNodePtr recv_node_ptr = graph_ptr->NewCNode(recv_input_list); in CreateRecvApplyKernel() local
1955 MS_EXCEPTION_IF_NULL(recv_node_ptr); in CreateRecvApplyKernel()
1958 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_builder.Build(), recv_node_ptr.get()); in CreateRecvApplyKernel()
1959 AnfAlgo::SetNodeAttr(kAttrEventId, MakeValue(event_id), recv_node_ptr); in CreateRecvApplyKernel()
1960 AnfAlgo::SetStreamId(stream_id, recv_node_ptr.get()); in CreateRecvApplyKernel()
1963 recv_node_ptr->set_abstract(abstract_none); in CreateRecvApplyKernel()
1964 return recv_node_ptr; in CreateRecvApplyKernel()