Was it the taser he picked up near the suspect and put on his belt? I couldn't tell on first look.
Looked smaller to me, like a pager or something. If the info about probes and wires is correct, it would have been a full-sized LE device, similar to what was on the ground at the point of the confrontation.