Searched refs:channel_dependency_group (Results 1 – 1 of 1) sorted by relevance
317 const HloComputation::ChannelDependencyGroup& channel_dependency_group, in ComputeInstructionPostOrder() argument357 if (channel_id && channel_dependency_group.count(*channel_id)) { in ComputeInstructionPostOrder()358 auto it = channel_dependency_group.find(*channel_id); in ComputeInstructionPostOrder()384 if (channel_id && channel_dependency_group.count(*channel_id)) { in ComputeInstructionPostOrder()385 auto it = channel_dependency_group.find(*channel_id); in ComputeInstructionPostOrder()397 ChannelDependencyGroup channel_dependency_group; in ComputeChannelDependencies() local402 channel_dependency_group[instruction->channel_id()].push_back( in ComputeChannelDependencies()408 channel_dependency_group[all_reduce_id.value()].push_back( in ComputeChannelDependencies()417 return channel_dependency_group; in ComputeChannelDependencies()421 auto channel_dependency_group = ComputeChannelDependencies(); in MakeInstructionPostOrder() local[all …]