Peter Hawkins 1a73743610 Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant. ...

This is a more descriptive name and a better location (next to other facilities for building XLA IR).

Quite a few users of the former xla_bridge.constant() didn't need anything other than uncanonicalized array constants. Change these users to use xla_client.ops.Constant instead; no need for the fancy utility in these cases.

PiperOrigin-RevId: 404270649

2021-10-19 08:40:51 -07:00

..

__init__.py

register the lax module for stack trace filtering

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

control_flow.py

Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.

2021-10-19 08:40:51 -07:00

fft.py

Improve real type conversion in a couple more places.

2021-10-18 13:50:11 -04:00

lax.py

Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.

2021-10-19 08:40:51 -07:00

linalg.py

Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.

2021-10-19 08:40:51 -07:00

other.py

Add missing typing.Optional type annotations to function parameters.

2021-05-27 20:10:23 -07:00

parallel.py

Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.

2021-10-19 08:40:51 -07:00

polar.py

Switch internal users of jax.ops.index_... to use x.at[x].set() APIs.

2021-09-13 19:48:29 -04:00