Searched refs:live_tensors_per_device (Results 1 – 1 of 1) sorted by relevance
169 std::unordered_map<string, std::deque<LiveTensor>> live_tensors_per_device; in InferFromTrace() local178 live_tensors_per_device[dev_stats.device()]; in InferFromTrace()235 &live_tensors_per_device[node_placement[input_node]]); in InferFromTrace()245 for (const auto& live_per_device : live_tensors_per_device) { in InferFromTrace()