jax
329321b0 - Add backend-specific lowering for cumsum/cumprod on TPU. (#2614)

Commit
5 years ago
Add backend-specific lowering for cumsum/cumprod on TPU. (#2614) * Add backend-specific lowering for cumsum/cumprod on TPU. Make cumsum/cumprod primitives so they can have backend-specific lowerings. * Disable cumulative reduction gradient test on TPU.
Author
Parents
Loading