U.S. weighs restricting Chinese investment in AI