The man arrested in the shooting at the White House correspondents' dinner, attended by US President Donald Trump, has been identified by US media as a mechanical engineer from California.