Using TF to compute Cholesky Decomposition gives error if matrix is any bigger than 2 by 2

Hi, guys!

I have been trying to find the cholesky decomposition of a bunch of randomly generated positive definite matrices but if the size of my matrix is any bigger than 2 by 2, tf gives me an error saying that the input is not correct for the tf.linalg.cholesky function.

I have even changed the data type to float64 but still no luck

Any help would be appreciated!