Searched refs:channel_id_map_ (Results 1 – 2 of 2) sorted by relevance
210 CHECK(channel_id_map_.find(channel_id) != channel_id_map_.end()); in GetChannel()211 return channels_[channel_id_map_.at(channel_id)]; in GetChannel()215 return channel_id_map_.find(channel_id) != channel_id_map_.end(); in HasChannel()318 TF_RET_CHECK(channel_id_map_.find(*hlo->all_reduce_id()) == in RecordInstructions()319 channel_id_map_.end()) in RecordInstructions()337 if (channel_id_map_.find(hlo->channel_id()) == channel_id_map_.end()) { in RecordInstructions()340 channel_id_map_[hlo->channel_id()] = channels_.size() - 1; in RecordInstructions()343 Channel& channel = channels_[channel_id_map_[hlo->channel_id()]]; in RecordInstructions()
274 absl::flat_hash_map<int64, int64> channel_id_map_; variable