Skip to content

Commit

Permalink
Update PyTorch export tutorial (openxla#2617)
Browse files Browse the repository at this point in the history
This PR updates the "Exporting StableHLO from PyTorch" tutorial.
Specifically:

- Update dependencies, combine the tensorflow dependency in the first
section to avoid warnings (about using tensorflow-cpu) on Kaggle
- Minor updates to narration and links, along with an introduction
statement
- Removed dynamic batch size code, addedd notes instead

(Supersedes openxla#2616, same updates but we get a rich diff here)

---------

Signed-off-by: Pavithra Eswaramoorthy <[email protected]>
  • Loading branch information
pavithraes authored Nov 11, 2024
1 parent 4d85f95 commit 619c326
Showing 1 changed file with 118 additions and 99 deletions.
Loading

0 comments on commit 619c326

Please sign in to comment.