In the US, customs and immigration processing at airports are both handled by CBP, and I've seen signs banning cell phone use at both passport control and customs areas.
That said, I'm also curious about the OP's question-- even if there might be reasons to ban their use, is this a published regulation somewhere?