US deputy attorney general Rod Rosenstein is heading to the White House expecting to be fired following reports that he had made comments critical of President Donald Trump, a source said.