Searched defs:rematerializable_map (Results 1 – 1 of 1) sorted by relevance
93 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in CanBeRematerialized()1359 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in AnyDenylistedOrNonRematerializable()1374 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map, in PickRematerializationCandidates()1752 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map, in RematerializeBestBlock()1857 absl::flat_hash_map<const HloInstruction*, bool> rematerializable_map; in RematerializeComputation() local