Searched refs:channel_instr (Results 1 – 2 of 2) sorted by relevance
140 const HloChannelInstruction* channel_instr = in NextChannelId() local142 if (channel_instr && channel_instr->channel_id()) { in NextChannelId()144 std::max(next_channel_id, *channel_instr->channel_id() + 1); in NextChannelId()
232 if (auto channel_instr = DynCast<HloChannelInstruction>(hlo)) { in CheckReplicaGroups() local233 if (channel_instr->channel_id()) { in CheckReplicaGroups()1543 auto channel_instr = DynCast<HloChannelInstruction>(instruction); in VerifyChannels() local1544 if (!channel_instr || !channel_instr->channel_id()) { in VerifyChannels()1547 channel_instructions[*channel_instr->channel_id()].push_back(instruction); in VerifyChannels()