Searched refs:live_tensor (Results 1 – 2 of 2) sorted by relevance
323 const auto& live_tensor = usage.live_tensors[i]; in PYBIND11_MODULE() local325 live_tensor.node, live_tensor.output_id, in PYBIND11_MODULE()326 live_tensor.memory_used, live_tensor.allocation_time.count(), in PYBIND11_MODULE()327 live_tensor.deallocation_time.count())); in PYBIND11_MODULE()
1041 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local1042 if (live_tensor.allocation_time > peak_time) { in IdentifySwappingCandidates()1043 peak_time = live_tensor.allocation_time; in IdentifySwappingCandidates()1050 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local1051 if (live_tensor.memory_used <= 1024) { in IdentifySwappingCandidates()1055 if (live_tensor.deallocation_time - live_tensor.allocation_time <= in IdentifySwappingCandidates()1058 VLOG(1) << "Not enough time to swap: skipping " << live_tensor.node; in IdentifySwappingCandidates()1062 if (skip_list->find(live_tensor.node) != skip_list->end()) { in IdentifySwappingCandidates()1066 graph.GetOutputPort(live_tensor.node, live_tensor.output_id); in IdentifySwappingCandidates()1072 mem_info.memory_used = live_tensor.memory_used; in IdentifySwappingCandidates()[all …]