Commit 5caa9870 authored by Anudhyan Boral's avatar Anudhyan Boral Committed by TensorFlower Gardener
Browse files

Add TensorFlow op to compute the LU decomposition of a matrix. 

For CPU, we use Eigen's PartialPivLU op, which has performance comparable to Cholesky decomposition. Unlike Cholesky this is also robust for non-SPD matrices.

For GPU, we use getrf and getrfbatched from cuSolver.

PiperOrigin-RevId: 224244390
parent f102c3c0
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment