Searched refs:RematerializationMode (Results 1 – 3 of 3) sorted by relevance
53 enum class RematerializationMode { enum88 RematerializationMode mode = RematerializationMode::kRecomputeAndCompress,190 RematerializationMode mode_;
499 HloRematerialization::RematerializationMode mode);776 HloRematerialization::RematerializationMode mode_;787 HloRematerialization::RematerializationMode mode) in MemoryUsageTracker()1407 HloRematerialization::RematerializationMode::kCompressOnly || in PickRematerializationCandidates()1408 mode_ == HloRematerialization::RematerializationMode:: in PickRematerializationCandidates()1441 if (mode_ == HloRematerialization::RematerializationMode::kCompressOnly) { in PickRematerializationCandidates()
56 HloRematerialization::RematerializationMode::kRecomputeAndCompress, in RunHloRematerialization()609 HloRematerialization::RematerializationMode::kCompressOnly, in RunHloRematerialization()