Searched refs:stream_infos_ (Results 1 – 1 of 1) sorted by relevance
522 auto it = stream_infos_.find(key); in AddStreamInfo()523 if (it != stream_infos_.end()) { in AddStreamInfo()534 stream_infos_.emplace(key, stream_info); in AddStreamInfo()605 stream_infos_.at(StreamKey(record.context, record.stream)); in SaveRecord()630 stream_infos_.at(StreamKey(record.context, record.stream)); in SaveRecord()704 absl::flat_hash_map<StreamKey, StreamInfo, hash<StreamKey>> stream_infos_; member in tensorflow::__anon18c3f2790211::CudaEventCollector