tree 4a34dcf819cd123c36a8cc8dd9d76d52bd0a75b4
parent 1acfde6e42c1007b028c82833c5922f14023d6fe
author Qiao Zhang <zhangqiaorjc@google.com> 1622588905 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1622589289 -0700

[JAX] Bug fix in TfrtCpuBuffer::logical_on_device_shape.

PiperOrigin-RevId: 376939192
Change-Id: I3f624c70cca924a66c7fdc5361cc6e863f26ccf0