Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_metadata.cc210 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()
Dhlo_module_group_metadata.h274 absl::flat_hash_map<int64, int64> channel_id_map_; variable