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