Searched defs:rematerializable_map (Results 1 – 1 of 1) sorted by relevance
97 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in CanBeRematerialized()1373 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in AnyDenylistedOrNonRematerializable()1388 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map, in PickRematerializationCandidates()1797 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map, in RematerializeBestBlock()1921 absl::flat_hash_map<const HloInstruction*, bool> rematerializable_map; in RematerializeComputation() local