Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc842 if (device.memory_usage > device.max_memory_usage) { in MarkCurrNodeExecuted()
843 device.max_memory_usage = device.memory_usage; in MarkCurrNodeExecuted()
918 int64 max_memory_usage = persistent_memory_usage + state.max_memory_usage; in Summary() local
920 max_memory_usage; in Summary()
929 << strings::HumanReadableNumBytes(state.max_memory_usage) in Summary()
930 << ", total = " << strings::HumanReadableNumBytes(max_memory_usage) in Summary()
973 max_memory_usage > 0 ? Round2(100.0 * op_mem_usage / max_memory_usage) in Summary()
1084 result[name] = state.max_memory_usage; in GetPeakMemoryUsage()
Dvirtual_scheduler.h117 int64 max_memory_usage; // Max temporary memory usage member
123 max_memory_usage = 0; in DeviceState()
Dvirtual_scheduler_test.cc2405 cpu_state.max_memory_usage); in TEST_F()
2437 cpu_state.max_memory_usage); in TEST_F()