mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/interpreters
History
Peter Hawkins ca17b6c08f Move functions out of xla.py closer to their users. 
Refactoring only, no changes intended. The goal is to shrink xla.py down to only its HLO-compatibility role, and remove things that aren't related to HLO compatibility.

Remove an unused top_k translation rule as well.

PiperOrigin-RevId: 554946059
2023-08-08 14:40:42 -07:00
..
__init__.py
Reapply: move jax.interpreters.ad to jax._src.interpreters.ad
2023-02-02 09:29:05 -08:00
ad.py
Apply pyupgrade --py39-plus.
2023-07-21 14:49:44 -04:00
batching.py
Apply pyupgrade --py39-plus.
2023-07-21 14:49:44 -04:00
mlir.py
Move functions out of xla.py closer to their users.
2023-08-08 14:40:42 -07:00
partial_eval.py
Apply pyupgrade --py39-plus.
2023-07-21 14:49:44 -04:00
pxla.py
Move functions out of xla.py closer to their users.
2023-08-08 14:40:42 -07:00
xla.py
Move functions out of xla.py closer to their users.
2023-08-08 14:40:42 -07:00