George Necula 0d8ffd33ab [shape_polyO] Improve handling of equality shape constraints ...

This fixes several bugs in presence of equality constraints where
the left-hand side is just a dimension variable.

First, such constraints were not applied when parsing variables.
Now, with a constraint `a == b` when we parse "a" we obtain `b`.

Second, when we evaluate symbolic dimensions that contain
dimension variables that are constrained to be equal to something
else, we may fail to find the dimension variable in the environment
because the environment construction has applied the constraints.
We fix this by looking up the unknown dimension variable in
the equality constraints.

Fixes: #23437
Fixes: #23456

2024-09-06 13:55:38 +03:00

..

export.md

Bump minimum jaxlib version to 0.4.31. The corresponding xla_extension_version is 279 and mlir_api_version is 57

2024-07-29 18:44:31 -07:00

index.rst

[export] Add documentation for jax.export

2024-06-12 19:44:47 +02:00

jax2tf.md

[shape_poly] Add documentation for shape polymorphism

2024-06-15 18:20:54 +03:00

shape_poly.md

[shape_polyO] Improve handling of equality shape constraints

2024-09-06 13:55:38 +03:00