Peter Hawkins def35b7e24 Remove scatter/gather dimension proto helpers. ...

These are unused since the MHLO switch.

PiperOrigin-RevId: 506969590

2023-02-03 12:40:31 -08:00

..

control_flow

Clean up: remove unused helper functions

2023-02-01 09:55:58 -08:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ann.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

convolution.py

Clean up: remove unused helper functions

2023-02-01 09:55:58 -08:00

eigh.py

[TPU] Add cutoff for nearly diagonal matrices in QDWH-eigh algorithm.

2022-11-23 08:17:50 -08:00

fft.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

lax.py

Clean up: remove unused helper functions

2023-02-01 09:55:58 -08:00

linalg.py

Clean up: remove unused helper functions

2023-02-01 09:55:58 -08:00

other.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

parallel.py

Update signature of ad.defbilinear to simplify transpose rules

2023-01-31 09:07:39 -08:00

qdwh.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

slicing.py

Remove scatter/gather dimension proto helpers.

2023-02-03 12:40:31 -08:00

stack.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

svd.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

utils.py

migrate internal dependencies from jax.core to jax._src.core

2022-12-16 21:00:14 -08:00

windowed_reductions.py

[jax2tf] Update the select_and_gather_add lowering for dynamic shapes

2022-12-19 14:44:10 +02:00