Searched refs:old_computation (Results 1 – 1 of 1) sorted by relevance
49 void MapComputation(const HloComputation* old_computation, in MapComputation() argument51 computations_[old_computation] = new_computation; in MapComputation()62 HloComputation* FindComputation(const HloComputation* old_computation) const { in FindComputation() argument63 return FindOrDefault(computations_, old_computation, nullptr); in FindComputation()72 HloComputation* GetComputation(const HloComputation* old_computation) const { in GetComputation() argument73 return FindOrDie(computations_, old_computation); in GetComputation()