Searched refs:after_bytes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_rematerialization.h | 46 int64 after_bytes; member |
| D | hlo_rematerialization.cc | 1293 sizes_->after_bytes = current_peak_memory; in Run() |