Searched refs:thread_local_set (Results 1 – 1 of 1) sorted by relevance
138 flat_hash_set<const HloComputation*> thread_local_set; in GatherComputationsByAllocationType() local146 bool in_thread_local_set = thread_local_set.contains(computation); in GatherComputationsByAllocationType()170 thread_local_set.insert(computation); in GatherComputationsByAllocationType()216 if (thread_local_set.contains(computation)) { in GatherComputationsByAllocationType()