Commit 1ddff940 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Add a Cholesky decomposition implementation.

Currently the implementation is fully unrolled, which can cause code size blowups at large matrix sizes. We can explore reducing code size in a subsequent change.

Create a new directory tensorflow/compiler/tf2xla/lib of XLA utility functions. Move batch matmul implementation into the utility directory. Add helpers for batch matmul, triangular solve, and Cholesky decomposition.

PiperOrigin-RevId: 175338698
parent dd58a908
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment