Google I/O returns May 18-20! Reserve space and build your schedule Register now


Make a function that tries Cholesky then the user-specified function.

Note this will NOT work under a gradient tape until b/177365178 is resolved. Also this uses XLA compilation, which is necessary until b/144845034 is resolved.

alternate_cholesky A callable with the same signature as tf.linalg.cholesky.
name Python str name prefixed to Ops created by this function. Default value: 'BackoffCholesky'.

run_backoff An function that attempts a standard Cholesky, and then tries alternate_cholesky on failure.