Searched refs:rematerialized_computations_ (Results 1 – 2 of 2) sorted by relevance
125 absl::flat_hash_set<const HloComputation*> rematerialized_computations_; variable
978 CHECK(!ContainsKey(rematerialized_computations_, computation)); in RematerializeComputation()1147 if (!ContainsKey(rematerialized_computations_, called_computation)) { in RematerializeComputation()1195 rematerialized_computations_.insert(computation); in RematerializeComputation()1210 rematerialized_computations_.clear(); in Run()