Improve error message reporting for check_numerics gradient.

At present the op message is only printed if the numeric check fails during
the op's 'forward' computation. If the check fails during the gradient, there is no
identifier on *which* op's gradient failed.

This is the Python equivalent of
https://github.com/tensorflow/tensorflow/commit/7e48bada

PiperOrigin-RevId: 222262823