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