Searched refs:rematerializable_map (Results 1 – 1 of 1) sorted by relevance
93 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in CanBeRematerialized() argument94 auto it = rematerializable_map->find(instruction); in CanBeRematerialized()95 if (it != rematerializable_map->end()) { in CanBeRematerialized()99 (*rematerializable_map)[instruction] = rematerializable; in CanBeRematerialized()566 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map,1359 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map) { in AnyDenylistedOrNonRematerializable() argument1364 if (!CanBeRematerialized(item->instruction, rematerializable_map)) { in AnyDenylistedOrNonRematerializable()1374 absl::flat_hash_map<const HloInstruction*, bool>* rematerializable_map, in PickRematerializationCandidates() argument1396 if (AnyDenylistedOrNonRematerializable(block, rematerializable_map)) { in PickRematerializationCandidates()1493 !CanBeRematerialized(next_item->instruction, rematerializable_map)) { in PickRematerializationCandidates()[all …]