Searched refs:module_output_size (Results 1 – 1 of 1) sorted by relevance
2023 int64 module_output_size = 0; in Run() local2026 [&module_output_size, module, this](const Shape& subshape, in Run()2028 module_output_size += size_function_(subshape); in Run()2032 memory_limit_bytes_ - module_output_size; in Run()2034 << HumanReadableNumBytes(module_output_size) in Run()2058 module_output_size; in Run()2086 module_output_size; in Run()