Searched refs:kAttrStreamId (Results 1 – 3 of 3) sorted by relevance
41 …AnfAlgo::SetNodeAttr(kAttrStreamId, MakeValue(reinterpret_cast<uintptr_t>(compute_stream)), kernel… in AssignGpuStream()52 …AnfAlgo::SetNodeAttr(kAttrStreamId, MakeValue(reinterpret_cast<uintptr_t>(comm_stream)), allreduce… in AssignGpuStream()193 uintptr_t send_stream = AnfAlgo::GetNodeAttr<uintptr_t>(mock_send_node, kAttrStreamId); in GenSendRecvCNodesForAllReduce()195 uintptr_t recv_stream = AnfAlgo::GetNodeAttr<uintptr_t>(mock_recv_node, kAttrStreamId); in GenSendRecvCNodesForAllReduce()
38 exec_node[kAttrStreamId] = stream_id_; in ExecNode2Json()
409 constexpr auto kAttrStreamId = "stream_id"; variable