Use a TPU-compatible approach to matrix inversion, until tf.linalg.inv is supported.
Use a TPU-compatible approach to matrix inversion, until tf.linalg.inv is supported.

PiperOrigin-RevId: 239801671

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor