Apple is set to raise its U.S. investment to $600 billion over the next four years, a White House official told CNBC.