U.S. to tell partners they must pick sides in AI race with China