George Necula 45ae4dfb9e [shape_poly] Remove caching for the symbolic shape evaluator ...

The caching used for the shape_poly.CachingShapeEvaluator leads to
leaked tracer errors. This is because the `lru_cache` is attached
to the `CachingShapeEvaluator.evaluate` and persists for the
duration of the program. It is possible to reimplement the caching,
but in this case caching does not help much so we just remove it.

2024-11-09 11:13:48 +02:00

..

__init__.py

[export] Move the export implementation to jax._src.export.

2024-06-09 08:59:50 -07:00

_export.py

[shape_poly] Remove caching for the symbolic shape evaluator

2024-11-09 11:13:48 +02:00

serialization_generated.py

[export] Add support for serialization for some custom PyTree nodes

2024-10-21 11:38:13 +02:00

serialization.fbs

[export] Cleanup in the export module.

2024-11-01 22:56:44 -07:00

serialization.py

[export] Cleanup in the export module.

2024-11-01 22:56:44 -07:00

shape_poly_decision.py

[shape_polyO] Improve handling of equality shape constraints

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

shape_poly.py

[shape_poly] Remove caching for the symbolic shape evaluator

2024-11-09 11:13:48 +02:00