U.S. weighs restrictions on China's access to AI memory chips