U.S. weighs restricting Chinese investment in artificial intelligence