Use tf.linalg.inv on TPU
Use tf.linalg.inv on TPU

PiperOrigin-RevId: 255117361

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