jax
4aee3810 - Disallow unreduced inputs for all primitives except those that implement the unreduced rule. Currently that's only `add`.

Commit
334 days ago
Disallow unreduced inputs for all primitives except those that implement the unreduced rule. Currently that's only `add`. PiperOrigin-RevId: 756902404
Author
Parents
Loading