[TF:XLA] Generalize existing support for keeping variables on an XLA device in...
[TF:XLA] Generalize existing support for keeping variables on an XLA device in reshaped form, instead allowing XLA devices to keep all tensors in a reshaped form outside an XLA computation. PiperOrigin-RevId: 196683444
Loading
Please sign in to comment