Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc2056 const int64 before_peak_memory = in Run() local
2060 << HumanReadableNumBytes(before_peak_memory); in Run()
2090 << HumanReadableNumBytes(before_peak_memory) << " (" in Run()
2091 << before_peak_memory << " bytes)"; in Run()
2092 const int64 reduced_peak_memory = before_peak_memory - current_peak_memory; in Run()
2098 sizes_->before_bytes = before_peak_memory; in Run()