Support for PSD kernels with structured inputs.

PiperOrigin-RevId: 476447997

Author: emilyfertig

tensorflow_probability/python/math/psd_kernels/BUILD

@@ -62,6 +62,7 @@ multi_substrate_py_library(
         # tensorflow dep,
         "//tensorflow_probability/python/internal:auto_composite_tensor",
         "//tensorflow_probability/python/internal:dtype_util",
+        "//tensorflow_probability/python/internal:nest_util",
         "//tensorflow_probability/python/internal:parameter_properties",
         "//tensorflow_probability/python/internal:slicing",
         "//tensorflow_probability/python/internal:tensorshape_util",