Searched refs:used_device_ids (Results 1 – 1 of 1) sorted by relevance
378 std::vector<bool> used_device_ids(bound_x * bound_y * bound_z * bound_core, in GetGeneralTPUExecutionDeviceAssignment() local406 if (used_device_ids[device_id]) in GetGeneralTPUExecutionDeviceAssignment()410 used_device_ids[device_id] = true; in GetGeneralTPUExecutionDeviceAssignment()