Searched refs:set_channel_id (Results 1 – 8 of 8) sorted by relevance
234 request.set_channel_id(GetChannelId(0)); in TEST_F()249 request.set_channel_id(GetChannelId(0)); in TEST_F()272 request.set_channel_id(GetChannelId(0)); in TEST_F()314 request.set_channel_id(GetChannelId(0)); in TEST_F()327 request.set_channel_id(GetChannelId(1)); in TEST_F()340 request.set_channel_id(GetChannelId(2)); in TEST_F()354 request.set_channel_id(GetChannelId(3)); in TEST_F()
2171 instr.set_channel_id(channel_id->handle()); in AllReduce()2370 send_instr.set_channel_id(handle.handle()); in SendWithToken()2377 send_done_instr.set_channel_id(handle.handle()); in SendWithToken()2422 recv_instr.set_channel_id(handle.handle()); in RecvWithToken()2430 recv_done_instr.set_channel_id(handle.handle()); in RecvWithToken()2468 send_instr.set_channel_id(handle.handle()); in SendToHost()2476 send_done_instr.set_channel_id(handle.handle()); in SendToHost()2508 recv_instr.set_channel_id(handle.handle()); in RecvFromHost()2517 recv_done_instr.set_channel_id(handle.handle()); in RecvFromHost()
494 next->set_channel_id(channel_id); in RewriteGraph()
1565 void set_channel_id(const absl::optional<int64>& channel_id);
369 void HloChannelInstruction::set_channel_id( in set_channel_id() function in xla::HloChannelInstruction379 proto.set_channel_id(*channel_id_); in ToProto()
220 void set_channel_id(const absl::optional<int64>& channel_id);
3770 void HloInstruction::set_channel_id(const absl::optional<int64>& channel_id) { in set_channel_id() function in xla::HloInstruction3771 return Cast<HloChannelInstruction>(this)->set_channel_id(channel_id); in set_channel_id()
143 void set_channel_id(int channel_id) { channel_id_ = channel_id; } in set_channel_id() function in cricket::TurnEntry1021 entry->set_channel_id(channel_id); in SetEntryChannelId()