tree cd2df398d1c9f3b30f697051a8b359cf6ee83d29
parent 0abdfe691e46da8e4dc47e2fbae57fdaca315fa7
author Skye Wanderman-Milne <skyewm@google.com> 1615597230 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1615597659 -0800

[XLA:Python] Add ProfilerSession bindings.

This allows for programmatically collecting a profiler trace that can be viewed in TensorBoard.

TESTING: I manually collected a JAX trace in a Cloud TPU VM. I'll add unit tests in a subsequent JAX change.
PiperOrigin-RevId: 362628356
Change-Id: I7e00bf0504572a364b8f91c01c3dda249ce21233