How to create XLA compiled model from a TF Saved Model

[ndeep27]

Issue type

Others

Have you reproduced the bug with TensorFlow Nightly?

No

Source

source

TensorFlow version

TF 2.11

Custom code

No

OS platform and distribution

No response

Mobile device

No response

Python version

No response

Bazel version

No response

GCC/compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current behavior?

NONE

Standalone code to reproduce the issue

I have a TF Saved Model which I want to convert to a CUDA Graph. I see that there is initial implmentation of CUDA Graphs in XLA here - https://github.com/tensorflow/tensorflow/commit/563c2ab50c43d16149ccfa143c8f97ebdb800a7e

How can I use XLA to convert a TF Saved Model to a CUDA Graph?

Relevant log output

No response