Searched refs:computation_cache_ (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | rng_bit_generator_expander.cc | 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()
|
| D | eigh_expander.h | 42 absl::flat_hash_map<string, HloComputation*> computation_cache_;
|
| D | cholesky_expander.h | 43 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
|
| D | qr_expander.h | 52 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
|
| D | triangular_solve_expander.h | 68 absl::flat_hash_map<string, HloComputation*> computation_cache_; variable
|
| D | rng_bit_generator_expander.h | 67 absl::flat_hash_map<RngGeneratorKey, HloComputation*> computation_cache_; variable
|
| D | cholesky_expander.cc | 232 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
|
| D | qr_expander.cc | 521 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
|
| D | eigh_expander.cc | 554 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
|
| D | triangular_solve_expander.cc | 578 computation_cache_.emplace(name, nullptr).first->second; in ExpandInstruction()
|