Searched refs:per_device_map_ (Results 1 – 2 of 2) sorted by relevance
202 : max_size_(max_size), per_device_map_(num_gpus) {} in AnnotationMap()220 absl::FixedArray<PerDeviceAnnotationMap> per_device_map_; variable
604 if (device_id >= per_device_map_.size()) return; in Add()605 auto& per_device_map = per_device_map_[device_id]; in Add()618 if (device_id >= per_device_map_.size()) return AnnotationInfo(); in LookUp()619 auto& per_device_map = per_device_map_[device_id]; in LookUp()