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.cc1254 const int64 before_peak_memory = in Run() local
1258 << HumanReadableNumBytes(before_peak_memory); in Run()
1284 << HumanReadableNumBytes(before_peak_memory) << " (" in Run()
1285 << before_peak_memory << " bytes)"; in Run()
1286 const int64 reduced_peak_memory = before_peak_memory - current_peak_memory; in Run()
1292 sizes_->before_bytes = before_peak_memory; in Run()