mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00

Adds lowering rules for reduce_all, reduce_any, reduce_min, and reductions to scalars. PiperOrigin-RevId: 650689871
Adds lowering rules for reduce_all, reduce_any, reduce_min, and reductions to scalars. PiperOrigin-RevId: 650689871