Searched refs:node_placement (Results 1 – 1 of 1) sorted by relevance
161 std::unordered_map<string, string> node_placement; in InferFromTrace() local164 node_placement[node_stats.node_name()] = dev_stats.device(); in InferFromTrace()235 &live_tensors_per_device[node_placement[input_node]]); in InferFromTrace()