xla
13526603 - Add function to generate stablehlo based callable from pytorch model (#5216)

Commit
2 years ago
Add function to generate stablehlo based callable from pytorch model (#5216) * Add function to generate stablehlo based callable from pytorch model Added function `torch_xla.experimental.stablehlo_saved_model.export_pytorch_model`. This function will take a pytorch Module and convert it into stablehlo bytecode.
Author
Parents
Loading