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