Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc1278 const int64 current_peak_memory = in Run() local
1282 << HumanReadableNumBytes(current_peak_memory) << " (" in Run()
1283 << current_peak_memory << " bytes), was " in Run()
1286 const int64 reduced_peak_memory = before_peak_memory - current_peak_memory; in Run()
1293 sizes_->after_bytes = current_peak_memory; in Run()
1298 if (current_peak_memory > memory_limit_bytes_) { in Run()
1303 HumanReadableNumBytes(current_peak_memory), current_peak_memory); in Run()