Check fail in densebincount
Thanks for the review @cantonios!

> Do we actually still get the check failure described in the original issue? I thought we fixed the work_element_count > 0 problem in general.

Yes the issue is still coming up - see this gist [here](https://colab.research.google.com/gist/madt2709/376f9f7aed33846fe0d76fe557fc2659/58178gpu.ipynb#scrollTo=zInMbWsqT7hy).
 > Thanks for the review @cantonios!
> 
> > Do we actually still get the check failure described in the original issue? I thought we fixed the work_element_count > 0 problem in general.
> 
> Yes the issue is still coming up - see this gist [here](https://colab.research.google.com/gist/madt2709/376f9f7aed33846fe0d76fe557fc2659/58178gpu.ipynb#scrollTo=zInMbWsqT7hy).

That gist passes with tf-nightly. Ah you're right! I should have checked that first. 

Well this makes this PR obsolete so will just close it and update the issue.

Thanks for the review (despite it being for nothing 😞 )
Added a test to reproduce the error from this issue https://github.com/tensorflow/tensorflow/issues/58178.

Second commit adds the relevant exception to the function.