Searched refs:thread_local_set (Results 1 – 1 of 1) sorted by relevance
132 flat_hash_set<const HloComputation*> thread_local_set; in GatherComputationsByAllocationType() local140 bool in_thread_local_set = thread_local_set.contains(computation); in GatherComputationsByAllocationType()164 thread_local_set.insert(computation); in GatherComputationsByAllocationType()210 if (thread_local_set.contains(computation)) { in GatherComputationsByAllocationType()