Searched refs:channel_group (Results 1 – 1 of 1) sorted by relevance
82 auto channel_group = computation->ComputeChannelDependencies(); in Build() local86 const auto add_input = [&channel_group, &inputs](HloInstruction* input) { in Build()89 auto it = channel_group.find(*input->all_reduce_id()); in Build()90 if (it != channel_group.end()) { in Build()109 auto it = channel_group.find(hlo->channel_id()); in Build()110 if (it != channel_group.end()) { in Build()122 auto it = channel_group.find(all_reduce_id.value()); in Build()123 if (it != channel_group.end()) { in Build()