Chris Jones 8d86a04727 [pallas] Allow TransformedRef to be passed to pl.load / pl.store, when idx = (). ...

PiperOrigin-RevId: 678257485

2024-09-24 08:17:21 -07:00

..

__init__.py

[Pallas TPU] Refactor ref indexers to transforms and support ref bitcast.

2024-09-15 17:53:29 -07:00

discharge.py

Clean up and fix primal type to tangent type mapping

2024-09-18 13:43:54 -07:00

indexing.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

primitives.py

[pallas] Allow TransformedRef to be passed to pl.load / pl.store, when idx = ().

2024-09-24 08:17:21 -07:00

types.py

Clean up and fix primal type to tangent type mapping

2024-09-18 13:43:54 -07:00

utils.py

[Pallas TPU] Refactor ref indexers to transforms and support ref bitcast.

2024-09-15 17:53:29 -07:00