Add missing dependency for //third_party/py/tensorflow_probability/python/math:diag_jacobian
Add missing dependency for //third_party/py/tensorflow_probability/python/math:diag_jacobian

PiperOrigin-RevId: 234195170

tensorflow/probability,tensorflow/probability