Searched refs:remat_able (Results 1 – 1 of 1) sorted by relevance
89 absl::flat_hash_map<const HloInstruction*, bool>* remat_able) { in CanBeRematerialized() argument90 auto it = remat_able->find(instruction); in CanBeRematerialized()91 if (it != remat_able->end()) { in CanBeRematerialized()95 (*remat_able)[instruction] = rematerializable; in CanBeRematerialized()865 absl::flat_hash_map<const HloInstruction*, bool>* remat_able) { in PickRematerializationCandidate() argument889 if (!CanBeRematerialized(candidate, remat_able)) { in PickRematerializationCandidate()993 absl::flat_hash_map<const HloInstruction*, bool> remat_able; in RematerializeComputation() local1032 memory_tracker, instruction_list, memory_limit_bytes, &remat_able); in RematerializeComputation()