Funding cuts to so-called prison riot squads are putting their safety at risk along with that of inmates and jail staff, Labour has claimed.