Searched refs:peak_usage_ (Results 1 – 2 of 2) sorted by relevance
56 auto it = peak_usage_.find(device); in GetPeakMemoryUsage()57 if (it == peak_usage_.end()) { in GetPeakMemoryUsage()74 std::unordered_map<string, MemoryUsage> peak_usage_; variable
65 for (const auto& peak_usage : peak_usage_) { in GetWorstCaseMemoryUsage()282 MemoryUsage& peak_mem_usage = peak_usage_[live_per_device.first]; in InferFromTrace()