Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc969 if (device.memory_usage > device.max_memory_usage) { in MarkNodeExecuted()
970 device.max_memory_usage = device.memory_usage; in MarkNodeExecuted()
1074 int64 max_memory_usage = persistent_memory_usage + state.max_memory_usage; in Summary() local
1076 max_memory_usage; in Summary()
1083 << ", peak = " << HumanReadableNumBytes(state.max_memory_usage) in Summary()
1084 << ", total = " << HumanReadableNumBytes(max_memory_usage) in Summary()
1145 max_memory_usage > 0 ? Round2(100.0 * op_mem_usage / max_memory_usage) in Summary()
1283 result[name] = state.max_memory_usage; in GetPeakMemoryUsage()
Dvirtual_scheduler.h132 int64 max_memory_usage; // Max temporary memory usage member
153 max_memory_usage = 0; in DeviceState()
Dvirtual_scheduler_test.cc2528 cpu_state.max_memory_usage); in TEST_F()
2566 cpu_state_0.max_memory_usage); in TEST_F()
2568 cpu_state_1.max_memory_usage); in TEST_F()
2602 cpu_state.max_memory_usage); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc3434 int64 max_memory_usage = 0; in VerifyAndExportHeapSimulatorTrace() local
3457 max_memory_usage = std::max(max_memory_usage, memory_usage); in VerifyAndExportHeapSimulatorTrace()
3460 VLOG(1) << "Max memory usage ignoring fragmentation: " << max_memory_usage; in VerifyAndExportHeapSimulatorTrace()