Searched refs:NextChannelId (Results 1 – 7 of 7) sorted by relevance
78 ag->channel_id() ? absl::make_optional(this->NextChannelId()) in RunOnComputation()99 next_channel_id_ = hlo_query::NextChannelId(*module); in Run()
36 int64 NextChannelId() { return next_channel_id_++; } in NextChannelId() function
3674 int64 next_channel_id = hlo_query::NextChannelId(*module); in Run()
82 int64 NextChannelId(const HloModule& module);
136 int64 NextChannelId(const HloModule& module) { in NextChannelId() function
433 hlo_query::NextChannelId(*(while_parent->parent())), in CreateSinkedAllReduces()
52 int64 next_channel = hlo_query::NextChannelId(*module); in ReplaceReplicatedAllReduce()