Searched defs:FusionInfoCache (Results 1 – 1 of 1) sorted by relevance
34 struct FusionInfoCache { struct38 void Invalidate(const HloInstruction* instr) { in Invalidate()45 absl::flat_hash_map<const HloInstruction*, int64_t> shared_memory_usage;46 absl::flat_hash_map<const HloInstruction*, int64_t> num_unnested_reductions;