Searched refs:per_hlo_cache (Results 1 – 3 of 3) sorted by relevance
277 std::unordered_map<HloInstruction*, PerHloCache> per_hlo_cache; member
349 auto& cache = state_.reshard_cache->per_hlo_cache[hlo()].reshard_cache; in Reshard()360 state_.reshard_cache->per_hlo_cache[resharded.hlo()] in Reshard()550 auto& cache = state_.reshard_cache->per_hlo_cache[hlo()].window_reshard_cache; in ReshardAsWindowedInput()824 auto& cache = state_.reshard_cache->per_hlo_cache[hlo()].reshard_cache; in Replicate()845 state_.reshard_cache->per_hlo_cache[resharded.hlo()] in Replicate()
830 state.reshard_cache->per_hlo_cache.erase(padded_slice_operand); in PartitionBaseCase()1082 state.reshard_cache->per_hlo_cache.erase(slice_operand); in PartitionBaseCase()