Tianjian Lu 020849076c [linalg] Add tpu svd lowering rule. ...

PiperOrigin-RevId: 445533767

2022-04-29 16:43:53 -07:00

..

__init__.py

register the lax module for stack trace filtering

2021-02-22 21:11:41 -08:00

ann.py

[MHLO] Remove most XLA translation rules.

2022-04-18 08:28:35 -07:00

control_flow.py

disable for now

2022-04-29 11:03:15 -07:00

convolution.py

[JAX] Delete last references to conv/dot translation rules.

2022-04-18 17:42:47 -07:00

fft.py

[MHLO] Remove most XLA translation rules.

2022-04-18 08:28:35 -07:00

lax.py

Merge pull request #10482 from lipracer:main

2022-04-29 08:37:08 -07:00

linalg.py

[linalg] Add tpu svd lowering rule.

2022-04-29 16:43:53 -07:00

other.py

Use real Precision type for lax.PrecisionType

2022-03-04 04:21:25 -08:00

parallel.py

Add unary xeinsum and allow named axis reductions for unary and binary xeinsums

2022-04-26 09:55:42 +00:00

polar.py

Document jax.lax.Precision

2021-11-08 14:15:31 -08:00

qdwh.py

[linalg] Fix type promotion in QDWH.

2022-03-03 20:59:28 -08:00

slicing.py

Change the default jnp.take mode to "fill".

2022-04-28 06:01:56 -07:00

svd.py

[linalg] Add tpu svd lowering rule.

2022-04-29 16:43:53 -07:00

utils.py

[MHLO] Remove most XLA translation rules.

2022-04-18 08:28:35 -07:00

windowed_reductions.py

add in mlir lowering for tokens

2022-04-21 11:28:58 -07:00