Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.cc82 auto channel_group = computation->ComputeChannelDependencies(); in Build() local
86 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()