Searched defs:live_tensor (Results 1 – 2 of 2) sorted by relevance
323 const auto& live_tensor = usage.live_tensors[i]; in PYBIND11_MODULE() local
1041 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local1050 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local