Searched refs:computation_cache_ (Results 1 – 8 of 8) sorted by relevance
60 auto it = computation_cache_.find(cache_key); in GetGeneratorComputation()61 if (it != computation_cache_.end()) { in GetGeneratorComputation()97 computation_cache_.emplace(cache_key, new_computation); in GetGeneratorComputation()
43 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
56 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
65 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
67 absl::flat_hash_map<RngGeneratorKey, HloComputation*> computation_cache_; variable
232 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
431 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
577 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()