Searched refs:gpu_device_ids (Results 1 – 1 of 1) sorted by relevance
268 std::vector<GlobalDeviceId> gpu_device_ids(local_device_states.size()); in BuildDistributedDevices() local283 gpu_device_ids[device_proto.local_device_ordinal()] = global_device_id; in BuildDistributedDevices()295 std::move(gpu_device_ids)); in BuildDistributedDevices()