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