xla
e7af313d - When exporting StableHLO to SavedModel, also include the original var (#5711)

Commit
1 year ago
When exporting StableHLO to SavedModel, also include the original var (#5711) * Pass bundle object to make_tf_function, so that it will include tf.variables as args to the XlaCallModule op. * fix format * When exporting StableHLO to SavedModel, also include the original variable tensor name from pytorch.
Author
Parents
Loading