Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc1221 int64 module_output_size = 0; in Run() local
1224 [&module_output_size, this](const Shape& subshape, in Run()
1226 module_output_size += size_function_(subshape); in Run()
1230 memory_limit_bytes_ - module_output_size; in Run()
1232 << HumanReadableNumBytes(module_output_size) in Run()
1256 module_output_size; in Run()
1280 module_output_size; in Run()