Sharad Vikram 907782d2e5 Deduplicate references closed over across branches of a lax.cond. ...

This fixes a correctness issue that could crop up when doing `run_state(cond)`.

PiperOrigin-RevId: 540795172

2023-06-15 23:58:14 -07:00

..

__init__.py

Add initial implementation of a run_state primitive

2023-04-03 21:32:32 -07:00

discharge.py

Deduplicate references closed over across branches of a lax.cond.

2023-06-15 23:58:14 -07:00

primitives.py

Verify that slices are trivial before discarding them in state primitives

2023-05-04 05:59:47 -07:00

types.py

Merge pull request #15149 from sharadmv:runstate

2023-04-04 10:56:25 -07:00

utils.py

Add initial implementation of a run_state primitive

2023-04-03 21:32:32 -07:00