Searched refs:rematerialized_computations_ (Results 1 – 2 of 2) sorted by relevance
173 absl::flat_hash_set<const HloComputation*> rematerialized_computations_; variable
1837 CHECK(!ContainsKey(rematerialized_computations_, computation)); in RematerializeComputation()1948 if (!ContainsKey(rematerialized_computations_, called_computation)) { in RematerializeComputation()1997 rematerialized_computations_.insert(computation); in RematerializeComputation()2012 rematerialized_computations_.clear(); in Run()