Home
last modified time | relevance | path

Searched refs:kCompressOnly (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.h55 kCompressOnly, // Only consider the kRecompute RematStrategy. enumerator
Dhlo_rematerialization_test.cc609 HloRematerialization::RematerializationMode::kCompressOnly, in RunHloRematerialization()
Dhlo_rematerialization.cc1407 HloRematerialization::RematerializationMode::kCompressOnly || in PickRematerializationCandidates()
1441 if (mode_ == HloRematerialization::RematerializationMode::kCompressOnly) { in PickRematerializationCandidates()