Searched refs:hlo_to_stream_number_ (Results 1 – 2 of 2) sorted by relevance
29 return hlo_to_stream_number_.contains(&hlo); in HasStreamAssigned()33 return FindOrDie(hlo_to_stream_number_, &hlo); in StreamNumberForHlo()42 InsertOrDie(&hlo_to_stream_number_, hlo, stream_num); in AssignStreamToHlo()
37 absl::flat_hash_map<const HloInstruction*, int> hlo_to_stream_number_; variable