Searched refs:peak_mem_usage (Results 1 – 2 of 2) sorted by relevance
282 MemoryUsage& peak_mem_usage = peak_usage_[live_per_device.first]; in InferFromTrace() local283 peak_mem_usage.used_memory = peak; in InferFromTrace()284 peak_mem_usage.live_tensors.clear(); in InferFromTrace()285 peak_mem_usage.live_tensors.reserve(live_at_peak.size()); in InferFromTrace()287 peak_mem_usage.live_tensors.push_back(*live); in InferFromTrace()
92 std::unordered_map<string, int64> peak_mem_usage = in Run() local94 for (const auto& mem_usage : peak_mem_usage) { in Run()