tree 76bbcbc7c0dceef348b54f01bf0cbb2140311935
parent 3ad4955f2d405168c0db6cb7e661c1a46f95eded
author A. Unique TensorFlower <gardener@tensorflow.org> 1624317250 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1624317558 -0700

[XLA:Python] Add a .live_buffers() on XLA:Python Device objects.

Adds an easy and public method to access the list of live buffers on a device from JAX.

PiperOrigin-RevId: 380686668
Change-Id: I76a885e50a9e77a0a3688b1cf4026ef6e35a6ffc