Searched refs:live_per_device (Results 1 – 1 of 1) sorted by relevance
245 for (const auto& live_per_device : live_tensors_per_device) { in InferFromTrace() local247 events.reserve(2 * live_per_device.second.size()); in InferFromTrace()248 for (const auto& live : live_per_device.second) { in InferFromTrace()283 MemoryUsage& peak_mem_usage = peak_usage_[live_per_device.first]; in InferFromTrace()