When it comes to white-ball cricket, Virat knows how to switch on and switch off, says Suresh Raina