Searched refs:StreamKey (Results 1 – 1 of 1) sorted by relevance
477 using StreamKey = std::pair<CUcontext, CUstream>; typedef in tensorflow::__anonee34c7620211::CudaEventCollector521 StreamKey key(context, stream); 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_;